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

并且在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

• 仔细检查您是否安装了启用了 cuda 的 pytorch，而不是 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。它适用于我安装 PyTorch 1.10 和 CUDA 10.2。

Answer 4

我迟到了，但对于通过 Google 来到这里的任何人来说，我通过按照本指南更新整个“堆栈”（GPU 驱动程序、CUDA、CuDNN、Anaconda、Pytorch）解决了类似的问题。