目录
1 问题
想安装一个torch+cuda,用的是官网的下载指令。但是一直都在报错:
Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple
Looking in links: https://download.pytorch.org/whl/torch_stable.html
Collecting torch==1.5.0+cu92
Downloading https://download.pytorch.org/whl/cu92/torch-1.5.0%2Bcu92-cp36-cp36m-win_amd64.whl (693.1MB)
0% | | 860kB 6.2kB/s eta 1 day, 7:10:03Exception:
Traceback (most recent call last):
... ...
ConnectionResetError: [WinError 10054] 远程主机强迫关闭了一个现有的连接。
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
... ...
pip._vendor.urllib3.exceptions.ProtocolError: ("Connection broken:
ConnectionResetError(10054, '远程主机强迫关闭了一个现有的连接。', None, 10054, None)",
ConnectionResetError(10054, '远程主机强迫关闭了一个现有的连接。', None, 10054, None))
2 原因分析
红得我心疼,下了很多次,刚开始还是90%断,后面0%就断了。查了度娘,是网络问题。(不明白啊,开热点也没办法搞定他)
3 解决方法
官网就给出了很棒的解决办法:
举个栗子:
① 在哪个步骤那里卡住,就直接复制里面的链接,比如红框里的。拷贝链接到浏览器下载。
其他版本的whl下载链接指路 官网whl诸多版本下载
② 打开cmd(或者,打开对应的虚拟环境)→cd到下载目录→pip install 安装下载文件
pip install XXX.whl
③ 测试:pip list
网上教程还有推荐清华镜像、conda安装等诸多方法,虽然我失败了,但是留一条路子,放到参考里。
4 torch+cuda安装教程(附)
cuda的下载:【软件安装】根据Tensorflow-GPU安装CUDA+CUDNN
torch官网下载:选择合适的版本,并复制最后一行代码到cmd中运行即可。
5 旧版本下载
我总是在为难我自己。。
注意电脑的GPU计算能力一定>=3.5。如何查看自己GPU的计算能力,官网查询
GPU计算能力<3.0的,没办法安装torch,肯定会报错(还没找到解决办法,只能安装cpu版本)。
5.1 问题
PyTorch no longer supports this GPU because it is too old.
5.2 解决办法
卸载torch→下载旧版本的pytorch的whl文件→再离线安装
官网给出了很多老版本的安装方式: 官网whl诸多版本下载 旧版本的torch
放几个windows相关版本链接:
py35
- cpu/torch-0.4.1-cp35-cp35m-win_amd64.whl
- cu80/torch-0.4.1-cp35-cp35m-win_amd64.whl
- cu90/torch-0.4.1-cp35-cp35m-win_amd64.whl
- cu92/torch-0.4.1-cp35-cp35m-win_amd64.whl
- cpu/torch-1.0.0-cp35-cp35m-win_amd64.whl
- cu80/torch-1.0.0-cp35-cp35m-win_amd64.whl
- cu90/torch-1.0.0-cp35-cp35m-win_amd64.whl
- cu100/torch-1.0.0-cp35-cp35m-win_amd64.whl
py36
- cpu/torch-0.4.1-cp36-cp36m-win_amd64.whl
- cu80/torch-0.4.1-cp36-cp36m-win_amd64.whl
- cu90/torch-0.4.1-cp36-cp36m-win_amd64.whl
- cu92/torch-0.4.1-cp36-cp36m-win_amd64.whl
- cpu/torch-1.0.0-cp36-cp36m-win_amd64.whl
- cu80/torch-1.0.0-cp36-cp36m-win_amd64.whl
- cu90/torch-1.0.0-cp36-cp36m-win_amd64.whl
- cu100/torch-1.0.0-cp36-cp36m-win_amd64.whl
py37
- cpu/torch-0.4.1-cp37-cp37m-win_amd64.whl
- cu80/torch-0.4.1-cp37-cp37m-win_amd64.whl
- cu90/torch-0.4.1-cp37-cp37m-win_amd64.whl
- cu92/torch-0.4.1-cp37-cp37m-win_amd64.whl
- cpu/torch-1.0.0-cp37-cp37m-win_amd64.whl
- cu80/torch-1.0.0-cp37-cp37m-win_amd64.whl
- cu90/torch-1.0.0-cp37-cp37m-win_amd64.whl
- cu100/torch-1.0.0-cp37-cp37m-win_amd64.whl
参考
win10下pytorch-gpu安装以及CUDA详细安装过程
使用conda/pip安装pytorch 0.3.1教程(mac/windos/linux)——有很多版本torch的下载码