RTX 3070 与 Pytorch 的兼容性答案

【问题标题】：RTX 3070 compatibility with PytorchRTX 3070 与 Pytorch 的兼容性
【发布时间】：2021-07-26 11:44:35
【问题描述】：

具有 CUDA 功能的 NVIDIA GeForce RTX 3070 sm_86 不兼容使用当前的 PyTorch 安装。当前的 PyTorch 安装支持 CUDA 能力 sm_37 sm_50 sm_60 sm_70。

所以我目前正在尝试训练神经网络，但我遇到了这个问题。看来我的GPU模型和我的PyTorch版本不兼容。

我的 nvcc -V 的输出是：

Cuda compilation tools, release 11.4, V11.4.48
Build cuda_11.4.r11.4/compiler.30033411_0

我的 PyTorch 版本是 1.9.0。

我尝试将 CUDA 版本从最初的 10 更改为 11.4，但没有任何变化。任何帮助将不胜感激。

【问题讨论】：

该问题与您的 CUDA 版本无关（尽管我怀疑您使用的 Pytorch 版本是否正确）。可能是 PyTorch 开发人员没有发布支持您的 GPU 的构建，因为它太新了。您的选择是等到它们支持您的 GPU 模型。或者如果可能的话，从源代码构建您自己的 PyTorch，并支持您的 GPU 模型
PyTorch compatibility matrix 建议可以为 CUDA 11.1 配置 pyTorch 1.9，这可能允许您使用 RTX 3070 运行。它说运行 conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge 并有注释conda-forge channel is required for cudatoolkit 11.1.
这可能是因为安装了错误的 cuda 版本的 pytorch。要使用计算能力 8.6 GPU，您必须安装 11.1 版本的 pytorch，因为计算能力 8.6 仅与 cuda >= 11.1 兼容。一旦你安装了正确的版本，如果你仍然有问题，请发布print(torch.cuda.version)和print(torch.cuda.get_arch_list())的结果

【解决方案1】：

这可能是因为您安装了带有 cuda==10.* 的 torch 包（例如 torch==1.9.0+cu102）。我建议尝试：

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

【讨论】：