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