ubuntu16.4安装cuda9.0+cudnn7.1

 一. 安装 cuda

下载cuda：https://developer.nvidia.com/cuda-toolkit-archive

 事前安装过驱动，所以不安装。

在.bashrc的末尾添加环境变量：

exportPATH=/usr/local/cuda-9.0/bin:$PATH

exportLD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
           

 使设置生效：

source ~/.bashrc

验证cuda是否安装成功

cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
           

 二. 安装cudnn

下载对应版本的cuDNN：https://developer.nvidia.com/rdp/cudnn-download

tar -zxvf cudnn-9.2-linux-x64-v7.6.1.34.tgz

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
           

验证安装：

nvcc -V
           

或者查看cuDNN版本验证：

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