欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Anaconda和pytorch在linux下安装教程

程序员文章站 2022-07-15 17:16:13
...

前几天,老板让我跑一个pytorch的代码,安装过程中遇到了很多的问题,不过最后一一解决了。
现在就Anaconda的安装和pytorch的安装做一下详细介绍;

Anaconda安装:

首先我们得确认我们所要安装的Anaconda的版本,我这里选择的是Anaconda3,安装包可以去Anaconda官网下载后传到linux上安装,或者可以在linux上输入wget https://repo.anaconda.com/archive/Anaconda3-2019.07-Linux-x86_64.sh。这样我们就完成了Anaconda安装包的下载。

然后在Anaconda3-2019.07-Linux-x86_64.sh的目录下,bash Anaconda3-2019.07-Linux-x86_64.sh(这里可以在输入Anaconda3-2019.07-Linux-x86_64.sh时输入Ana后按tab键就可以自动打出整个安装包的名字)。然后回车之后就进入Anaconda的安装过程了,中间遇到一路yes,在让你确认Anaconda的安装路径的时候,可以根据你自己的选择选择安装路径或者直接回车选择默认的安装路径。

在安装完anaconda之后,我们还需配置环境变量。我们只需输入

echo 'export PATH="/root/userfolder/anaconda3/bin:$PATH"' >> ~/.bashrc
source ~/.bashrc

即可。

在配置好环境变量之后,我们输入conda,假如没报错,那就说明我们的Anaconda安装成功了。耶耶耶耶!!!!
下一步就将要进入到我们的pytorch的安装了。

pytorch安装

首先我们得将清华镜像添加到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/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/

进入pytorch官网,根据系统、python版本、cuda版本选择下载命令。
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
**ps:**这里有一个注意的点:添加镜像源之后,安装pytorch的过程还是很慢,甚至中断退出安装,这里的原因是:
我们在pytorch官网获得的下载命令:conda install pytorch torchvision cudatoolkit=9.0 -c pytorch中的 - c pytorch参数指定了conda获取pytorchchannel,在此指定为conda自带的pytorch仓库。
因此,只需要将-c pytorch语句去掉,就可以使用清华镜像源快速安装pytorch了。