一开始用的清华源安装pytorch1.5.1,试了好几次都是下到一般卡住了,最后下载失败,pip
安装也是只有十几k的速度。最终使用豆瓣的源下载成功,而且速度非常的快,值得一试。
进入pytorch官网
依照官网提供的安装命令,安装有GPU加速的pytorch=1.5.1
版本,torchvision
选择0.6.1
的,然后在后面加个豆瓣的源就行了。
pip install torch===1.5.1 torchvision===0.6.1 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple