具有 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，但没有任何变化。任何帮助将不胜感激。