我正在尝试在 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）的人，我将此作为一个单独的问题发布。