我想在 Google Compute Engine 上使用 GPU 运行自定义 Docker 映像。我已构建映像并将其推送到 Google Container Registry。将 Container-Optimized OS 用于 Google Cloud Engine 中的主机似乎是合乎逻辑的,因为除了 Docker、Nvidia GPU 驱动程序和nvidia-container-runtime
.
我设法用这个解决方案安装了 nvidia-drivers 。
--gpu all
但是如果没有 nvidia-container 运行时,我无法使用 GPU(使用选项)运行我的 Docker 映像。此步骤在官方 Docker 文档中指定。
有没有办法在 Google Cloud VM 的 Container-Optimized OS 上安装 nvidia-container-runtime?