我试图在一个附带 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在这个问题的答案回答了我的问题，但是我没有在我的搜索中找到这个问题，因为他们的问题涵盖了更广泛的范围，而我的问题特定于CUDA运行时/驱动程序错误。我将其作为单独的问题发布，供那些遇到此错误但不知道其是什么症状（即忘记使用nvidia-docker）的人使用。
nvidia-docker只是最后的答案。可能主要是因为该答案是在问题提出之后很长时间才编写的。 - BlameTheBits