linux（ubuntu和centos）配置pytorch

---

一、安装anaconda

1.安装anaconda

在anaconda官网下载.sh安装包

在下载路径下安装anaconda

bash Anaconda3-2020.07-Linux-x86_64.sh

一路回车直到，是否接受协议？yes
安装目录选择默认目录则直接回车，若自定义目录则输入路径。
是否初始化？yes （后面可以省去conda命令不生效，修改bashrc的步骤）

2.配置路径

安装完成后默认配置路径需要输入如下命令行

source ~/.bashrc

关掉终端窗口并重启，输入conda list命令验证

二、安装pytorch

1.创建pytorch虚拟环境

使用anaconda安装tensorflow，caffe、pytorch等框架，可以先通过anaconda创建虚拟环境，然后在不同的虚拟环境下装上不同的版本的框架（例如在一个虚拟环境上安装一个tensorflow1.4，在另一个虚拟环境中上安装一个tensorflow，在另一个虚拟环境中安装pytorch）。并且如果配置出错，直接删除虚拟环境的包即可，而不影响这个系统。
代码如下：

 conda create -n pytorch python=3.7

运行结果

2.激活pytorch虚拟环境

 source activate pytorch

激活后，会看到前面会显示环境名字比如（pytorch），如下

3.在虚拟环境下安装pytorch

打开pytorch起始页，根据所使用Linux服务器的信息，进行选择：

选择完后，在命令行执行上图红中的命令即可

 conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

4.验证是否安装成功

执行下面命令行验证

python  # 进入python 环境
import torch  # 导入安装的pytorch包
torch.cuda.is_available()  # 检查cuda是否可以使用

输出结果

输出True即为成功。