如何利用Anaconda和Conda安装PyTorch

  • 1.前言
  • 2.如何安装
  • 2.1 下载并安装Anaconda
  • 2.1.1 从清华镜像下载Anaconda
  • 2.1.2编译所下载的文件
  • 2.1.3 刷新环境
  • 2.1.4 解决Anaconda没有被添加到环境里问题
  • 1)确定Anaconda3文件的路径
  • 2)添加Anaconda到系统环境中
  • 3)判断添加是否成功
  • 2.2 进入PyTorch网站得到运行命令的正确命令
  • 2.3 确定环境配置是否冲突


1.前言

在公用服务器上注册了一个新的账号,然而系统里面空空如也,想在服务器上跑一个用到依赖PyTorch的神经网络代码,因此首先需要配置PyTorch。

2.如何安装

2.1 下载并安装Anaconda

我是从清华镜像下载Anaconda,毕竟直接从官网下下载真的慢,没办法。当然这步可以到Anaconda的官网上直接下载(如果你能打开并且下载速度飞快),我的系统是Linux的,所以下载的是Linux版本的。如果是window系统的用户就很方便直接双击下载好的安装包,安装就好了,这里不赘述如何在win下安装anaconda。接下来主要说明如何在linux安装Anaconda,由于linux需要在终端下输入安装指令,具体安装过程如下:

2.1.1 从清华镜像下载Anaconda

在桌面处直接打开终端(Open in terminal)输入命令:

curl -O https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2019.10-Linux-x86_64.sh

可以看到我安装的版本是Anaconda3-2019-10-Linux-x86_64不喜欢这个版本的朋友可以到https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive看看喜欢哪个版本就下哪个,系统是window的也能下注意一下后缀说明是win还是Linux,下载对应的系统版本就行。

2.1.2编译所下载的文件

从执行上一个命令可以发现桌面多了一个.sh文件,那就是从镜像里下载得到的安装包。之后继续在终端输入命令:

bash Anaconda3-2019.10-Linux-x86_64.sh

注意bash后面输入的其实是刚才所下载的文件的名字,如果你下载的版本不同记得修改成相应的文件名,这样就可以编译.sh文件开始安装。

这个过程可能是持续一会,安装了一会发现没反应的话就按一直按回车直到出现提示。如下所示:

conda环境中查看Pytorch 版本_CUDA


窗口提示你:Do you accept the license terms [yes|no]

输入:yes 再回车

等待安装,最后安装成功。

2.1.3 刷新环境

安装期间窗口会提示运行conda在bashrc中添加路径,安装好后,由于~/.bashrc中新加了东西,需要输入命令刷新一下系统环境配置:

source ~/.bashrc

然而我在安装的时候压根没提醒是否在bashrc中添加路径,害。安装都安装好了就只能老老实实地把Anaconda添加到环境里了。以下方法适用和我一样没有成功将Anaconda添加到环境里的朋友。那么如何判断你有没有成功将Anaconda添加到环境里,很简单,你在终端里输入命令:
首先输入:

source ~/.bashrc

再输入:

conda

得到的提示是

conda:command not found

那么恭喜你!没有将Anaconda添加到环境里,那么你就需要按照以下的方法来将Anaconda添加到环境里。

2.1.4 解决Anaconda没有被添加到环境里问题

1)确定Anaconda3文件的路径

刚才安装Anaconda3成功后,你的系统会有一个文件夹命名为Anaconda3,双击打开Anaconda3可以看到里面有一个bin文件和lib文件,请找到它们的位置并记住它们的路径,比如我的路径就是

/home/user12/Anaconda3/bin

2)添加Anaconda到系统环境中

因此打开终端输入命令:

echo 'export PATH=/home/user12/Anaconda3/bin/:$PATH'>>~/.bashrc
echo 'export LD_LIBRARY_PATH=/home/user12/Anaconda3/lib/:$LD_LIBRARY_PATH'>>~/.bashrc
source ~/.bashrc
3)判断添加是否成功

上一步成功将Anaconda添加到环境中并刷新环境,再继续输入命令

conda

发现conda指令可以被识别出来了,那你就成功了。

2.2 进入PyTorch网站得到运行命令的正确命令

其实第一步安装Anaconda就已经解决了很多问题,第一步完成后python3.7是自动安装了,如果没有自动安装的朋友也可以用conda命令安装python。
个人推荐:其实在终端下载有很多麻烦,有时候下着下着就崩了。还有一个安装Pytorch的方式就安装PyCharm,然后直接在PyCharm里安装所需要的所有库。
当然第二种方式就是在终端里安装:直接通过PyTorch官网下载安装速度比较慢,甚至经常会崩掉,因此可以选择通过清华源来安装。清华源镜像还是比较稳的,一般没什么问题。

进入PyTorch官网:https://pytorch.org/

conda环境中查看Pytorch 版本_bash_02


选择所需要的版本要求生成相应的安装指令

在终端接入清华镜像源软链接:

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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

安装PyTorch:

conda install pytorch torchvision cudatoolkit=10.1

注意去掉官网命令后面的-c pytorch 安装完成。

2.3 确定环境配置是否冲突

以下我的运行环境要求为例:
OS:Linux
Package Manager :conda
Python:3.7
CUDA:10.1(由于我是在公共服务器上运行我的代码,因此CUDA是已经提前安装好的,因此这里省略了CUDA的安装,需要安装的朋友可以自行搜索如何安装CUDA)

老实说我的PyTorch装了好几个版本了,如果你的代码没问题但报错,那么你可以看看是不是cuda和PyTorch版本不兼容的问题。有的时候PyTorch版本太高,显卡带不动,得装低版本的PyTorch。