我试图在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)的人。
问题是我用docker
运行容器,而不是nvidia-docker
。使用nvidia-docker
运行docker容器修复了问题。