Pytorch指定版本安装，基于Anaconda

我习惯用anaconda建虚拟环境，就以anaconda安装pytorch1.3为例

1、安装Anaconda

下载链接：https://www.anaconda.com/

下载好默认安装即可

2、创建虚拟环境

（1）开始菜单打开Anaconda Prompt，如下图

 （2）在打开的命令行窗口（如下图）中，输入 

conda create -n pytorch1.3 python=3.6

敲回车（ps:其中，“pytorch1.3”为你定义的虚拟环境名，什么名都可以；python为3.6版本）

（3）输入 activate pytorch1.3, 进入虚拟环境，上图的(base)变为(pytorch1.3)就表示成功进入。

 3、在虚拟环境中安装pytorch1.3

 （1）提前下载torch和torchvision安装包，这种安装方式更简单，不用镜像也不用担心网速问题。

下载链接：https://download.pytorch.org/whl/torch_stable.html

安装pytorch实际上就是安装torch和torchvision两个包，这两个包相互依赖，且版本要一一对应，还要和GPU版本对应，具体对应关系如下图：（以下图片来自网络）

cuda和torch对应关系：

 所以我们要下载的包为：torch1.3.0 和 torchvision0.4.1

 在下载链接中找到对应的文件，如下图：

（ps:文件名中包含很多信息，cu101表示cuda版本为10.1 ；cp36表示python版本为3.6 ；win表示windows系统，根据自己的实际情况找到对应包即可，没有gpu的就下载cpu开头的）

 （2）下载好之后开始安装

在第2步的第（3）步中，已经进入了虚拟环境，直接使用以下命令安装torch：

（ps：要先定位到你下载的包所在的路径，否则无法找到包）

pip install torch-1.3.0-cp36-cp36m-win_amd64.whl

torch-1.3.0-cp36-cp36m-win_amd64.whl就是上一步下载好的包名。

然后安装torchvision，和torch是一样的。

（3）检查是否安装成功以及cuda版本是否匹配

首先在虚拟环境中输入python，进入编译环境，输入以下代码

import torch
print(torch.cuda.is_available())

输出 True 则安装成功

关于cuda和cudnn的安装这里不再赘述，注意版本关系。

计算机视觉学习交流可以私聊我进群