Question

我正在尝试使用Google Compute尝试一些深度学习模型。我希望使用Nvidia的CUDA容器来运行这些模型。一种相对便宜的方法是利用Google的容器优化操作系统来托管容器。

当我转到驱动程序时，未安装Nvidia驱动程序。由于它是专有操作系统，因此我似乎找不到找到为Google提供的任何GPU安装任何驱动程序的方法。

我发现最接近的东西是：https://github.com/ContainerEngine/accelerators/tree/master/cos-nvidia-gpu-installer，但这意味着创建一个新的容器，而不是使用一些GPU加速库附带的Nvidia。

Answer 1

这可能是您想要的： https://github.com/GoogleCloudPlatform/cos-gpu-installer#how-to-use

您可以尝试运行此命令，看看它是否适合您的用例：

gcloud compute instances create $USER-cos-gpu-test \
--image-family cos-stable \
--image-project cos-cloud \
--accelerator=type=nvidia-tesla-k80 \
--boot-disk-size=25GB \
--maintenance-policy=TERMINATE \
--metadata-from-file "cos-gpu-installer-env=scripts/gpu-installer-env,user-data=install-test-gpu.cfg,run-installer-script=scripts/run_installer.sh,run-cuda-test-script=scripts/run_cuda_test.sh"

如何轻松地为Google的容器优化操作系统安装Nvidia驱动程序？

1 个答案: