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

Question

最近，我开始使用docker图像。我想部署基于PyTorch的文本分类模型，这需要图形处理器来运行。当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运行就可以了。