RTX 3070 与 Pytorch 的兼容性

Question

具有 CUDA 功能 sm_86 的 NVIDIA GeForce RTX 3070 与当前 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，但没有任何变化。任何帮助将不胜感激。

Answer 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