解决Ubuntu安装pytorch过慢的问题

2023-01-01 22:14:58

最近在学习pytorch需要在服务器搭建一个安装pytorch框架，所以结合我这几天的学习经历写了一个在Ubuntu16.04版环境下安装pytorch。

1.安装anaconda

anaconda是一个很好用的python包管理器，对于经常使用python是非常友好的，可以在anaconda下新建不同的环境，以满足不同开发环境的需求。anaconda的安装这个网上有很多教程，不做具体的的描述。

2.建一个anaconda环境

这是新建一个名字为pytorchgpu的环境，python版本为3.6，可根据需要更改

conda  create -n pytorchgpu python=3.6

激活环境

source activate pytorchgpu

如果想要退出这个环境，可以使用下面的命令

source deactivate pytorchgpu

3.修改conda的下载源

pytorch最近已经更新到了稳定版本的1.7，从Pytorch官网上可以看到有多种安装方式：

比较常见的方法是通过pip和conda安装，当使用conda安装的时候，可能会遇到下载过慢的问题，尤其是文件：pytorch包文件，下载的时候可能会遇到无尽的等待，甚至失败（亲身体会）。这里推荐用清华源替代默认conda源的方法，可以解决下载过慢的问题。因此需要添加国内的下载源，我使用的清华的镜像：

清华conda源地址：https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/

添加清华源

输入下面的命令：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

为了保险起见，我还添加了其他的源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

4.下载pytorch

pytorch的下载要在之前新建的环境下安装，不然就安装到其他环境中了，容易混乱。
登录到pytorch的官网，查看下载的命令。
pytorch的官网：https://pytorch.org/get-started/locally/
我的cuda版本为9.2，系统为Ubuntu，所以选择如上，官网给了下载命令，但是要进行修改。

conda install pytorch torchvision cudatoolkit=9.2 -c pytorch

这里要注意一定不能使用上面的命令，我们要对其进行修改为如下所示，就是删掉 ”-c pytorch“

conda install pytorch torchvision cudatoolkit=9.2

之后就是慢慢等待包的下载和安装。

5. 验证是否安装成功

查看pytorch是否安装成功：

python
import torch
torch.rand(2,2)

torch.__version__
#查看pytorch版本

如果上面的程序运行无误，就表示安装成功。接下来就可以使用该环境去跑新的代码。

码农公寓