按照我自己的经验,用 pip3从官网安装还是比较靠谱的,官方源(-c pytorch)、清华镜像源经常中断。
pip3 install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
以上是一个示例,可以根据你自己的需要替换对应的版本就可以了。
如果指定的版本不存在,就会报错,提示你有哪些版本可以提供下载使用。