docker 容器中的 GPU 用于深度学习任务

Question

最近，我开始使用 docker 图像。 我想部署基于 PyTorch 的文本分类 model ，这需要 GPU 才能运行。 当调用 docker 映像时，它无法在 VM 中检测到 GPU。 因此，我的代码因抛出no Cuda device found error而失败。

这是我FROM gcr.io/deeplearning-platform-release/pytorch-gpu.1-0的基础镜像。 我不知道在 docker 中安装 Nvidia 驱动程序的所有步骤。 请帮帮我。

Answer 1

将 Nvidia 驱动程序安装到容器中并没有帮助，因为 GPU 设备没有暴露给容器。 而是使用nvidia-docker运行。