pytorch - 断言错误 : Torch not compiled with CUDA enabled (depite several reinstallations)

Question

每当我尝试在 pytorch 中将变量移动到 cuda(例如 torch.zeros(1).cuda())时，我都会收到错误消息“AssertionError:Torch 未在启用 CUDA 的情况下编译”。此外，torch.cuda.is_available() 返回 False。

我已经阅读了几个解决此错误的答案，但出于某种原因，多次尝试重新安装 cuda 和 putorch 并没有改变任何东西。以下是我使用的一些设置:

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

conda install pytorch torchvision cudatoolkit=11 -c pytorch-nightly

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

但同样的错误仍然存​​在。可能是什么问题？

一些设置:我使用的是 Ubuntu 20.04，GPU 是 RTX 2080，nvidia-smi 工作正常(NVIDIA-SMI 460.91.03，驱动程序版本:460.91.03，(最大可能)CUDA 版本:11.2)

Answer 1

尝试使用 pip 安装pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html/p>

你可以通过这个线程来获得详细的解释 Pytorch for cuda 11.2