GPU 不适用于 Pytorch

Question

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

>>> 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

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

__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

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