python 安装 pytorch的超详细步骤

首先贴上pytorch的官网：https://pytorch.org/

 

不多介绍，直接开始安装步骤：

1、打开https://pytorch.org/get-started/locally/

2、选择对应要安装的版本。

第一行表示稳定版和预览版

第二行表示系统

第三行选择哪种安装方式

第四行语言

第五行CUDA（看你的显卡支不支持）如果没有就选None。

如何查看支持的CUDA的版本，看这篇文章：https://blog.csdn.net/he99774/article/details/104713315

最后一行，打开cmd就可以安装了。

 

在cmd安装速度奇慢无比，只有几kb每秒，800M的文件要下好几天。

所以我是直接打开https://download.pytorch.org/whl/torch_stable.html

找到对应的版本，打开我的vpn，用浏览器下。

我是下的这两个（自己下要选择对应的python版本和cu还是cpu）

https://download.pytorch.org/whl/cu101/torch-1.4.0-cp35-cp35m-win_amd64.whl

https://download.pytorch.org/whl/cu101/torchvision-0.5.0-cp35-cp35m-win_amd64.whl

然后用pip安装whl就好了。

 

国内的镜像我没找到，就这样了。