pytorch安装与配置

为方便用户安装使用，PyTorch 官方提供了多种安装方法。本节将介绍两种常用的安装方式，读者可以根据自己的需求选用。

1.使用conda安装

conda是Anaconda自带的包管理器。如果使用Anaconda作为Python环境，则除了使用pip安装，还可使用conda进行安装。同样，在PyTorch官网中（https://pytorch.org/）选择操作系统、包管理器conda、Python版本及CUDA版本，对应不同的安装命令。我们以在Windows下安装Python3.6、CPU 版本的PyTorch为例介绍。如下图所示。

安装命令如下:

>conda install pytorch torchvision cpuonly -c pytorch

conda的安装速度可能较慢，建议国内用户，尤其是教育网用户把conda源设置为清华tuna。在命令行输人如下命令即可完成修改。

>conda config --add channels https://mirrors. tuna. tsinghua. edu. cn/ anaconda/pkgs/free/
>conda config--set show_ channel urls yes

使用anaconda安装

需要安装pytorch和torchvision安装包，pytorch-cpu可能只含有cpu版本吧（因为我已经下载好pytorch安装包的缘故，所以没有pytorch安装包，仅演示不必下载）

conda常用命令

打开Anaconda Prompt即可

conda install 网址(或文件路径+文件名)

2.使用pip安装

目前，使用pip安装PyTorch二进制包是最简单、最不容易出错，同时也是最适合新手的安装方式。从PyTorch官网(http:/pytorch.org/)选择操作系统、包管理器pip、Python版本及CUDA版本，会对应不同的安装命令。

以Windows平台、pip 安装、Python3.6及CPU为例，安装命令如下(根据不同系统配置，可将pip改为pip2或pip3 )。

>pip install torch==1.3.1+cpu torchvision==0.4.2+cpu -f https://download.pytorch.org/whl/torch_stable.html
>pip install torchvision

安装好PyTorch之后，还需安装Numpy，安装命令如下。

>pip install --upgrade numpy

或者使用系统自带的包管理器(apt, yum等)安装numpy,然后使用pip升级。
 

>apt install python-numpy

>pip install一 upgrade numpy

pip使用清华源加速下载pytorch最新版本

上述方法也是从官网上直接下载pytorch，所以也会产生下载速度过慢的问题。

>python3 -m pip install --upgrade torch torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple

网上找的可以试一试，总结就是使用清华源下载速度会快一点，不过清华源有时候会关闭，所以有时候可能需要将pytorch环境下载下来进行安装（将相关网址改为路径名+文件名）。本人采用conda安装方式。

---

全部安装完成后，打开Python,运行如下代码。

import torch
print(torch.__version__)

没有报错则表示PyTorch安装成功。

安装过程中需要注意以下几点。

( 1 ) PyTorch对应的Python包名为torch而非pytorch。

(2)若需使用GPU版本的PyTorch,需要先配置英伟达显卡驱动，再安装PyTorch。

参考：深度学习框架pytorch入门与实践