【问题记录】NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch。

服务器上跑代码的时候遇到了报错:

NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.

错误是当前3090显卡与torch的版本不支持。
解决办法就是更改torch版本,3090显卡用以下命令语句即可解决:

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

你可能感兴趣的:(pytorch,人工智能,python)