我正在尝试使用GPU加速器在Google计算引擎上运行一个tensorflow容器。

尝试过命令

gcloud compute instances create-with-container job-name \
  --machine-type=n1-standard-4 \
  --accelerator=type=nvidia-tesla-k80 \
  --image-project=deeplearning-platform-release \
  --image-family=common-container \
  --container image gcr/io/my-container \
  --container-arg="--container-arguments=xxxx"

但是得到警告

  

警告：此容器部署机制需要经过容器优化的OS映像才能工作。从cos-cloud项目中选择一个映像（成本稳定，cos-beta，cos-dev映像系列）。

我还尝试了cos-cloud项目中的系统映像，该项目似乎没有CUDA驱动程序，因为tensorflow记录了警告cuInit failed。

想知道在具有GPU支持的Google计算引擎上运行tensorflow容器的正确方法是什么吗？