如何快速地安装正确的pytorch版本

一.遇到的问题

pytorch的安装可以在官网https://pytorch.org/中找到对应的命令：

下图就是官网中指定版本生成的命令，此时pytorch是1.4，操作系统是Windows，使用 pip命令进行安装，对应的底层语言平台是python，cuda是用于GPU的包版本，此时生成的命令是

pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html

 

但是在实际的安装中发现，我在anaconda prompt下使用该pip命令安装时下载速度特别慢，还老是在中途断掉，这是因为pytorch官网是外网。我指定其他国内镜像，比如清华镜像，中科大镜像，阿里云镜像都不行。阿里云显示找不到这个包，清华、中科大镜像显示连接超时，这可能是版本太新，这些镜像里面还没有的缘故。不过其他老的版本我也没有使用这些镜像试试能不能安装成功。出现错误的页面如下所示：

对应的错误为：

File "c:\users\skj\anaconda3\lib\site-packages\pip\_vendor\urllib3\response.py", line 307, in _error_catcher
    raise ReadTimeoutError(self._pool, None, 'Read timed out.')
pip._vendor.urllib3.exceptions.ReadTimeoutError: HTTPSConnectionPool(host='download.pytorch.org', port=443): Read timed out.

二.解决方法

1.先在官网生成想要版本的命令,比如：

pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html

2. 在anaconda prompt等安装的命令行等地方执行上面的命令，会生成自己电脑上python版本对应的包的下载地址，比如上面的第二张图中进度条的正上方有这样一句：

Downloading https://download.pytorch.org/whl/cpu/torch-1.4.0%2Bcpu-cp35-cp35m-win_amd64.whl (77.4MB)

我们复制地址https://download.pytorch.org/whl/cpu/torch-1.4.0%2Bcpu-cp35-cp35m-win_amd64.whl到我们的浏览器中，科学上网或者使用VPN连接到外网进行下载，我的VPN速度很快，达到5M/s，比命令行下安装快很多。

3.下载完直接使用pip，调用本地的whl文件安装pytorch

比如上面的包我下载之后放在我的电脑目录：C:\MySoftware\pythonWHL\torch-1.4.0+cpu-cp35-cp35m-win_amd64.whl

直接使用命令pip install C:\MySoftware\pythonWHL\torch-1.4.0+cpu-cp35-cp35m-win_amd64.whl

至此，安装成功。

参考链接：

https://blog.csdn.net/a790209714/article/details/78708945/

https://blog.csdn.net/sinat_30281495/article/details/88546242