RTX 3070与Pytorch的兼容性

Question

NVIDIA GeForce RTX 3070与CUDA功能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，没有任何变化。如有任何帮助，我将不胜感激。

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