AssertionError：Torch 未在启用 CUDA 的情况下编译（尽管进行了多次重新安装）

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

您可以通过此线程获取 Pytorch for cuda 11.2 的详细说明