可在线安装,也可以离线安装,推荐使用离线安装方法,快!!!
1.使用清华源安装库
pip install pytorch==1.5.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
2.离线安装
到链接pytorch下载地址下载对应版本的pytorch和torchvision
我用的是linux版本,python3.7,pytorch1.5.1和torchvision0.6.1
torchvision0.6.1
下載完成之后,直接安装
pip install /home/jsj/Downloads/torch-1.5.1-cp37-cp37m-linux_x86_64.whl
pip install /home/jsj/Downloads/torchvision-0.6.1-cp37-cp37m-linux_x86_64