Docker容器中的Cuda运行时/驱动程序不兼容

Question

我正在尝试在Pytorch随附的Docker容器中运行此简单的代码行。

import torch
torch.cuda.set_device(0)

我收到此错误：

RuntimeError: cuda runtime error (35) : CUDA driver version is insufficient for CUDA runtime version at torch/csrc/cuda/Module.cpp:32

运行torch.cuda.is_available()返回False。

主机具有最新的Nvidia驱动程序。 Pytorch随Cuda一起提供，因此应该不存在不兼容问题。

什么可能导致此问题？

编辑： @Patel Sunil对this question的回答回答了我的问题，但是我在搜索中没有遇到这个问题，因为它们的问题很广泛，而我的问题特定于cuda运行时/驱动程序错误。对于那些遇到此错误但不知道这是什么症状的人（即，忘记使用nvidia-docker），我将其作为一个单独的问题发布。

Answer 1

问题是我使用docker而不是nvidia-docker运行容器。使用nvidia-docker运行docker容器可解决此问题。