笔者因为实验需要,不同的模型需要不同版本版本的pytorch,于是安装pytorch就总是出现一个问题,那就是断。比如需要安装torch1.4版本,使用Anaconda Prompt尝试了几次:
pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple
总是失败,最后怒了,直接使用离线包安装。。。。。。
具体教程:
- 进入PyTorch官网
https://pytorch.org/
:
- 接着点
Get Started
:
- 找到下载地址(见附图中标红部分) 根据第一步得到的CUDA选择不同版本进行下载
此处有两种方法:
直接在电脑端访问此网址,找到需要的下载包(不推荐,速度太慢了)
采用接下来介绍的方法
把刚才复制的链接发送给手机(通过微信/QQ都可以)
- 在手机端复制链接至浏览器访问该地址,寻找需要下载的文件(我的是torch1.4)点击即可下载
- 最后,使用命令行:
pip install torch-1.4.0-cp36-cp36m-win_amd64.whl
注:如果是虚拟环境的话,需要将下载后的whl文件放在Scripts文件夹里面,再使用上述命令: