RTX 3070与火炬的兼容性
提问于 2021-07-26 03:44:35
具有CUDA功能的
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版本，但并没有改变。任何帮助都将是非常感谢的。
回答 1
Stack Overflow用户
发布于 2022-01-27 22:40:48
这可能是因为您已经安装了带有cuda==10.*的火炬包(例如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页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/68529258
复制相关文章
点击加载更多
腾讯云开发者
