服务器上跑代码的时候遇到了报错:
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