GPU 不适用于 Pytorch答案

【问题标题】：GPU is not available for PytorchGPU 不适用于 Pytorch
【发布时间】：2021-10-24 05:20:52
【问题描述】：

我今天安装了 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

任何建议都会有所帮助

编辑：我正在使用从their website 上生成的命令安装的 PyTorch 1.10。使用python 3.9.7。我还在新的 conda 环境中再次安装了 PyTorch，但遇到了同样的问题。

【问题讨论】：

您是否尝试过使用较旧的 CUDA 版本（例如 10.1 / 10.2）安装 Pytorch？如果您的平台支持 11，那么旧版本也应该支持...
您安装了一个未使用 CUDA GPU 加速构建的 PyTorch 版本。您需要安装不同版本的 PyTorch。在 CUDA 加速构建中，torch.version.cuda 将返回一个 CUDA 版本字符串。在非 CUDA 构建上，它返回 None
打开终端，输入nvidia -smi查看nvidia驱动是否安装成功。
@talonmies 感谢您指出这一点。更改我的 PyTorch 版本似乎已经解决了这个问题。我真的很感激！

标签： python pytorch anaconda conda

【解决方案1】：

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

【讨论】：

并不是 Pytorch `1.10 不支持 CUDA——它确实支持。是你安装的build没有编译CUDA支持
我安装了主页上列出的构建，但它没有工作。我选择了我正在使用 CUDA 并且没有其他构建选项
可能是他们搞砸了，并且不小心将非 GPU 构建分发为 GPU 构建。

【解决方案2】：

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

【讨论】：

【解决方案3】：

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

【讨论】：