Ubuntu16.04 Anaconda+Pytorch安装

[前提] 已安装python3.7

一、Anaconda安装

Anaconda下载地址: link.

选择Python 3.7 版的Anaconda进行下载：Anaconda3-5.3.0-Linux-x86_64.sh。

下载完成后，在文件所在目录打开终端，添加执行权限。

chmod 755 Anaconda3-5.3.0-Linux-x86_64.sh 

执行安装命令。

bash Anaconda3-5.3.0-Linux-x86_64.sh

安装过程中遇到询问，直接 yes 就行。

执行到最后，看到 Thank you for installing Anaconda3! 就已经安装成功了。

二、下载pytorch、torchvision、cudatoolkit安装包

pytorch：[link.] pytorch-1.7.1-py3.7_cuda11.0.221_cudnn8.0.5_0.tar.bz2
torchvision：[link.] torchvision-0.8.2-py37_cu110.tar.bz2
cudatoolkit：[link.] cudatoolkit-11.0.221-h6bb024c_0.tar.bz2

三、anaconda下创建 python3.7 环境

终端 source 一下 .bashrc。

source ~/.bashrc

创建名为 pytorch_py37 的 python3.7 工作环境。

conda create -n pytorch_py37 python=3.7

创建完成后，输入下面命令测试环境是否能激活。激活成功后，用户名前会多出创建的工作环境名称，比如这里是(pytorch_py37)。

conda activate pytorch_py37

四、安装pytorch、torchvision、cudatoolkit

激活 pytorch_py37 工作环境的前提下，依次执行以下命令安装pytorch、torchvision、cudatoolkit。

conda install --offline pytorch-1.7.1-py3.7_cuda11.0.221_cudnn8.0.5_0.tar.bz2

conda install --offline torchvision-0.8.2-py37_cu110.tar.bz2 

conda install --offline cudatoolkit-11.0.221-h6bb024c_0.tar.bz2

五、补全缺少的依赖库

执行完成以上步骤，pytorch、torchvision、cudatoolkit 已成功安装到 anaconda pytorch_py37 工作环境中，但相关的依赖库如 numpy 等一大堆依赖库仍然需要安装。但是，如果一项一项安装的话就太麻烦了，好在 anaconda 可以自动补全，这时候使用 anaconda 的好处就体现出来了。

conda install pytorch torchvision cudatoolkit=11.0

这样就安装成功了。

六、测试pytorch

pytorch_py37 环境中运行 python。

执行测试语句。

import torch
x = torch.rand(5, 3)
print(x)

SUCCESS && END