问题是由于当前显卡与torch版本有冲突。
以我自己为例:当前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
问题是由于当前显卡与torch版本有冲突。
以我自己为例:当前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