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

Question

最近，我开始使用docker图像。我想部署基于PyTorch的文本分类模型，它需要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运行。