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

我试图在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\u available（）

返回

False

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

什么会导致这个问题

编辑：

---

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

问题是我使用

docker

运行容器，而不是

nvidia docker

。使用nvidia docker运行docker容器修复了问题。

我很高兴您找到了问题的解决方案！但这不是一个好主意吗？嗯，

nvidia docker

只是最后一个答案。可能主要是因为答案是在提问后写的。编辑我的帖子以澄清-谢谢！