ubuntu 16.04下安装pytorch配置caffe2（cuda 9.0+cudnn 7.6）

caffe2已经合并进啦pytorch里面，所以我们可以通过安装pytorch完成对caffe2的配置，相较于安装caffe2源码的方式，操作上更为简单。

• 操作系统: Ubuntu 16.04.5 LTS
• 显卡：GTX 1050TI
• 显卡驱动版本: 384.130
• CUDA Toolkit版本: CUDA Toolkit 9.0
• cuDNN 版本: cuDNN v7.6.0
• pytorch 版本: pytorch-1.3.1

1、安装cuda 9.0

1.1 下载安装

点击进入cuda的官方下载地址，在列表中选择下载“CUDA Toolkit 9.0 (Sept 2017)”。如下图选择Base Installer进行下载。
下载完毕后在存放“cuda_9.0.176_384.81_linux.run”安装文件的文件夹下新建终端，执行以下命令完成cuda的安装。

sudo sh cuda_9.0.176_384.81_linux.run

安装过程中会进行多次询问，注意当问你“Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81?
”时，因为我已经装好了显卡驱动，所以选择n，其余的选择默认就行。

1.2 设置环境变量

打开bashrc文件

sudo gedit ~/.bashrc

在bashrc文件最末尾添加下面两行，保存后退出

export PATH="/usr/local/cuda-9.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH"

使修改生效

source  ~/.bashrc

1.2 安装验证

使用如下命令进行验证，安装成功后就会输出cuda的路径。

echo $PATH 和 echo $LD_LIBRARY_PATH

2、安装cudnn 7.6

2.1 下载安装

点击进入cudnn的官方下载地址直接点击login，在弹出的登陆界面点击最下方的“Login with your social account >”即可通过使用QQ登录来实现免注册登录下载。

下载完成之后解压文件，执行如下命令把文件拷贝到CUDA的目录下完成安装。

sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64

2.2 安装验证

使用下面命令查看cudnn版本，完成验证，若正确安装会输出如下图

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

3、安装Anaconda 3

3.1 下载安装验证

通过点击清华镜像网站的网址，快速下载Anaconda的安装包，下载完成后，在安装包文件夹下打开终端输入命令完成安装。具体版本自己修改。

bash ./Anaconda3-5.3.1-Linux-x86_64.sh

安装过程中问你需不需要安装Vscode时，如果用不着就选no，其余的默认就好。

3.2 添加环境变量

打开bashrc文件。

sudo gedit ~/.bashrc

在文件末尾处添加以下语句，保存后退出。

export PATH=/home/XXX/anaconda3/bin:$PATH    XXX为自己的用户名

安装完成后新建终端后输入pyton，可知此时默认的python由python2变成了python3

3.3 换源实现高速下载

Anaconda默认下载源，下载速度贼慢，可通过换成国内的清华源实现加速下载。

具体实现参考：https://blog.csdn.net/qq_22059843/article/details/105952288

4、安装caffe

4.1 安装pytorch

谨慎起见，我选择新建环境变量

 conda create -n caffe2 pip python=3.7.1

激活环境变量

source activate caffe2

安装cuda 9对应版本的pytorch

conda install pytorch torchvision cudatoolkit=9.0

4.2 安装其他必要依赖项目

选择使用pip加速下载

pip install -i https://pypi.douban.com/simple/ protobuf
pip install -i https://pypi.douban.com/simple/ future

4.3验证

可以通过以下语句验证是否安装成功;

python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"