检查cuda版本
nvidia-smi
查看pytorch版本
pip list
conda list
版本过低
这里最好去pytorch的官网查看自己需要哪款pytorch
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
参考官网
检查cuda版本
nvidia-smi
查看pytorch版本
pip list
conda list
版本过低
这里最好去pytorch的官网查看自己需要哪款pytorch
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
参考官网