请参考:https://www.cnblogs.com/tryhardwy/p/14659131.html
注意:直接粘贴
pip install torch==1.8.1+cu102 torchvision==0.9.0+cu102 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
,会提示找不到torchvision==0.9.0+cu102 导致安装失败!
正确的安装语句是:
pip install torch==1.8.1+cu102 torchvision==0.9.1+cu102 -f https://download.pytorch.org/whl/torch_stable.html