Pytorch安装

Pytorch在Ubuntu和Windows上安装的差别不大，下面将以Windows上安装作为示例。

1. 选择Pytorch版本
   进入Pytorch官网，查找所需版本的Pytorch安装命令： （1）Pytorch Build是最新的版本1.5；
   （2）系统是Windows；
   （3）安装包的类型：因为是在Python下进行开发，所以选择Conda，也可以选择Pip，如果进行C++开发的话，可以选择LibTorch，下载编译好的链接库；
   （4）语言：还是要根据开发需要选择，大部分人应该都用Python进行开发；
   （5）Cuda版本：这个非常关键！Cuda版本一定要和系统安装的Cuda版本一致，不然可能会出现各种奇怪的错误。如果已经安装了一个Cuda版本，但是在安装选项中又没有对应Cuda版本的Pytorch版本，可以选择安装多个Cuda版本，然后设置环境变量进行Cuda版本的切换。不过我也没有在Windows上做过这种尝试，因为大部分开发是在Ubuntu上进行的，关于Ubuntu上多个版本Cuda版本切换，这里简要说明一下方法：建立软连接，然后将软连接路径添加到环境变量中，通过改变软连接连接的Cuda版本进行版本切换。
2. 安装Pytorch
   打开anaconda prompt
   (1) 创建虚拟环境：

conda create -n pytorch-1.5

(2) 激活创建的虚拟环境：

activate pytorch-1.5

(3) 输入1中得到的安装命令：

conda install pytorch torchvision cpuonly -c pytorch

下载过程会比较慢，网上有一些快速安装的方法，但是我在尝试的时候发现效果都不明显，也可以考虑直接下载对应版本的安装包，然后在anaconda中进行安装，不过没有直接使用命令行来得简单方便。

最后在prompt终端中输入python，然后按照下图输入代码，如果安装成功的话，就可以看到你安装的pytorch版本：