我正在尝试使用 Google Compute 来玩一些深度学习模型。我希望使用 Nvidia 的 CUDA 容器来运行这些模型。一种相对便宜的方法是利用 Google 的容器优化操作系统来托管容器。
当我转到驱动程序时,未安装 Nvidia 驱动程序。由于它是专有操作系统,我似乎找不到为 Google 提供的任何 GPU 安装任何驱动程序的方法。
我发现的最接近的是:https ://github.com/ContainerEngine/accelerators/tree/master/cos-nvidia-gpu-installer ,但这意味着创建一个新容器而不是使用一些 GPU 加速库附带的 Nvidia 容器.