linux下使用anaconda安装pytorch

pytorch

本文主要介绍一下使用anaconda在在linux环境下的安装pytorch的流程。由于pytorch安装比较简单，这里只写出一些主要的流程。anaconda的安装在此就不再赘述。

1.确定自己的系统版本

因为pytorch在0.4已经不支持比较老的系统版本了。以下是pytorch的member在github上的原话。

In 0.4, we deprecated RHEL6/CentOS6 and Ubuntu 12.04 support.

因此，可以先查看一下自己的系统版本是否支持当前的pytorch版本(0.4),我用的是CentOS,使用下面的命令

cat /etc/redhat-release

其他的可以参考

https://www.cnblogs.com/parrynee/archive/2010/05/16/1736652.html
https://blog.csdn.net/zhuying_linux/article/details/6859286

2.确定自己的CUDA版本

如果没有CUDA的可以忽略这条。CUDA的安装大家可以自行谷歌百度，网上很多教程。
CUDA版本获取：

cat /usr/local/cuda/version.txt

cudnn版本获取:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 

3.建立虚拟环境

conda create -n pytorch (python=2.7/3.6)

这个指令是建立一个名为pytorch的虚拟环境，以后所有的操作都在这个虚拟环境中，这可以避免各个版本的python或者框架之间的相互干扰。括号中的内容可以让你指定python的版本，如果不输入括号中的内容，系统会安装目前最新的python版本。

虚拟环境建立好了之后使用如下语句激活虚拟环境

source activate pytorch

这样一句之后，我们可以看到在命令行的最前面会出现(pytorch)表示我们已经进入了这个虚拟环境了。以后所有的操作都在这个虚拟环境下面进行。

如果需要退出该虚拟环境，可以使用如下命令。

source deactivate

如果在配置的过程中发生了什么问题，可以先退出虚拟环境然后移除该虚拟环境，再重新配置

conda remove -n pytorch --all

4.安装pytorch

进入pytorch官方网站获取安装指令

https://pytorch.org/
在官网主页根据你的系统和CUDA，python版本，选择conda安装方式。我的是

conda install pytorch torchvision -c pytorch

5.测试

根据提示安装好了之后可以python然后import torch尝试是否安装成功了。

6.问题

如果遇到了以下问题：

ImportError: /usr/lib64/libstdc++.so.6: version `CXXABI_1.3.8' not found (required by /export/home/dyh/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/lib/libATen.so)

说明了在/usr/lib64/路径下没有libstdc++.so.6这个文件

这个文件可以在当前虚拟环境的路径下找到，因此可以在.bashrc文件中添加下面一行

export LD_LIBRARY_PATH=/export/home/dyh/anaconda3/envs/pytorch/lib:$LD_LIBRARY_PAT

把anaconda3的路径更改成自己的路径就可以了。
然后source .bashrc
然后可以尝试退出/重启之后再重新测试

如果还有这个问题可能是gcc版本不对，把gcc版本升级到7以后即可。

conda uninstall gcc
conda install -c conda-forge isl=0.17.1
conda install -c creditx gcc-7