Linux安装配置pytorch教程

约定

脚本等文件在/home/fan下的Documents文件夹内

下载的各种包，在/home/fan下的Download文件夹内

数据在/data/fan下

一、下载配置anaconda

不要下载miniconda，下载miniconda3

 

下载地址如下：

清华镜像：https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/

官网：https://docs.conda.io/en/latest/miniconda.html

 

直接用命令

先进入你要下载的目录下

cd /home/fan/Download

再执行下载命令 

wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda-latest-Linux-x86_64.sh

然后执行安装

bash Miniconda3-latest-Linux-x86_64.sh

然后一直enter和yes即可

 

最后验证

conda --version

如果不成功，推出shell重新连接

 

二、创建pytorch专属环境

创建名为 pytorch的环境

conda create -n pytorch python=3.7

每次进入都要用以下命令才能使用，同理，安装各种包也要先进入该环境

(base) fanjh@917-02:~$ conda activate pytorch
(pytorch) fanjh@917-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/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --set show_channel_urls yes | cat ~/.condarc

 

安装pytorch

先确定本机cuda的版本，没有安装cuda和cudnn的自行百度安装即可，这里不赘述

cat /usr/local/cuda/version.txt

我这里是：CUDA Version 10.2.89，所以在  https://pytorch.org/get-started/locally/  中，我选择的命令是

 

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

但不要加最后的-c pytorch，否则不会使用国内镜像源，因此应该执行

conda install pytorch torchvision torchaudio cudatoolkit=10.2

但有时候无法调用GPU，默认下载的是1.4版本，所以这里可以改用pip安装pytorch

pip install torch torchvision

 

三、安装各种包

要么conda install ，要么pip install

安装cv2

pip install opencv-python

安装