pytorch安装与配置

妖狐艹你老母 2023-06-14 10:39 57阅读 0赞

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

### 1.使用conda安装 ###

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

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3RvdWdoX3N0cm9uZw_size_16_color_FFFFFF_t_70][]

安装命令如下:

>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安装**

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3RvdWdoX3N0cm9uZw_size_16_color_FFFFFF_t_70 1][]

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

**conda常用命令**

打开Anaconda Prompt即可

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

### 2.使用pip安装 ###

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

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3RvdWdoX3N0cm9uZw_size_16_color_FFFFFF_t_70 2][]

以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入门与实践

[https_pytorch.org]: https://pytorch.org/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3RvdWdoX3N0cm9uZw_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20191119104124231.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3RvdWdoX3N0cm9uZw==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3RvdWdoX3N0cm9uZw_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/20191207121546258.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3RvdWdoX3N0cm9uZw==,size_16,color_FFFFFF,t_70
[http_pytorch.org]: http://http/pytorch.org/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3RvdWdoX3N0cm9uZw_size_16_color_FFFFFF_t_70 2]: https://img-blog.csdnimg.cn/20191119110106956.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3RvdWdoX3N0cm9uZw==,size_16,color_FFFFFF,t_70