pip3安装pytorch
1、进入官网下载相应的pytorch
2、离线下载轮子
可以直接复制Run this Command中的命令进行安装,但由于安装速度实在太慢,更换了源亦是如此。所以,为了加快安装速度,事先下载wheel文件,进行离线安装。
浏览器打开上图红框网站。
使用迅雷下载,对应的版本。
3、使用pip3安装whl文件
4、安装pytorch
由于已经安装好whl文件,所以再次使用pip3命令安装将非常快。
pip3 install torch===1.7.1 torchvision===0.8.2 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
注意,在此步骤中会安装一些numpy、feature等packet,会影响一些安装速度,甚至出现一些错误。所以,可以选择提前安装这些packet。(提示缺什么安装什么)
5、测试
import torch if __name__ == '__main__': print(torch.__version__) print(torch.cuda.is_available())
1.7.1
True
Process finished with exit code 0