GPU不可用于Pytorch

Question

我今天安装了Anaconda, CUDA和PyTorch，我无法在torch中访问我的GPU (RTX 2070)。我遵循了所有的安装步骤，PyTorch工作正常，否则，但当我试图访问GPU无论是在shell或脚本我得到

>>> import torch
>>> torch.cuda.is_available()
False
>>> torch.cuda.device_count()
0
>>> print(torch.version.cuda)
None

运行conda list显示这个是我安装的包

cudatoolkit               11.3.1               h59b6b97_2

，在conda环境中运行numba -s显示

__CUDA Information__
CUDA Device Initialized                       : True
CUDA Driver Version                           : 11030
CUDA Detect Output:
Found 1 CUDA devices
id 0    b'NVIDIA GeForce RTX 2070'                              [SUPPORTED]
compute capability: 7.5
pci device id: 0
pci bus id: 1
Summary:
1/1 devices are supported

，ok测试全部通过。CUDA 11.3是PyTorch和我的GPU支持的计算平台之一，这是我安装的版本。

我已经尝试重新安装CUDA，我在Windows 10上，nvcc --version显示CUDA已安装Build cuda_11.3.r11.3/compiler.29745058_0

任何建议都会有帮助的

编辑:我使用PyTorch 1.10从他们的网站上生成的命令安装。使用python 3.9.7。我还在一个新的conda环境中重新安装了PyTorch，并遇到了同样的问题。

Answer 1

• 仔细检查你安装的pytorch是否启用了cuda而不是CPU版本
• 打开一个终端，运行nvidia-smi，看看它是否检测到你的GPU。再次检查你的Cuda版本是否与PyTorch要求的版本相同。如果您使用的是旧版本的Cuda，请下载最新版本。

Answer 2

将CUDA降级到10.2并使用PyTorch LTS 1.8.2可以让PyTorch现在使用GPU。根据@talonmies的评论，PyTorch 1.10似乎不支持CUDA

Answer 3

如果使用conda，请尝试更新conda。我可以在CUDA 10.2下安装PyTorch 1.10。