ubuntu16.04安装cuda、cudnn

以cuda_9.0.176.384.81_linux.run为例。

1.安装cuda

 下载cuda：CUDA Toolkit Archive | NVIDIA Developer

  注意cuda对应的驱动版本号，电脑上安装的驱动版本必须高于此版本，如cuda_9.0.176.384.81_linux.run，对应可使用的驱动版本号要高于384.81。

   进入cuda.run所在目录，修改程序权限

   sudo chmod 777 cuda_9.0.176.384.81_linux.run

   安装：sudo ./cuda_9.0.176.384.81_linux.run

  

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

 

 

  

  

添加环境变量：

sudo gedit ~/.bashrc 

末尾添加

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

export LD_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

安装cuda10.2时accept后弹出选择框，x号代表选定安装，去掉x号后就不安装，然后选定install回车安装。

2.安装cudnn

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

    解压cudnn包（以7.6为例）

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*

 官网教程：Installation Guide :: NVIDIA Deep Learning cuDNN Documentation

验证安装：

nvcc -V

或者查看cuDNN版本验证：

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