linux16.04安装cuda和cudnn

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/TH_NUM/article/details/82804360

安装cuda

wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.0.176-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1604_9.0.176-1_amd64.deb
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda-9-0 -y
sudo nvidia-xconfig

配置环境路径：
vim ~/.bashrc

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/lib/:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME="/usr/local/cuda"
export PATH="$CUDA_HOME/bin:$HOME/bin:$HOME/.local/bin:/usr/bin:/usr/local/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/openmpi/lib/:$LD_LIBRARY_PATH"

执行 source ~/.bashrc

重启机器之后，nvidia-smi生效

安装cudnn

在https://developer.nvidia.com/rdp/cudnn-archive 下载适合的版本的cudnn:
在16.04 x86版本我下载的是：cudnn-9.0-linux-x64-v7.1.tgz

tar -xvf cudnn-9.0-linux-x64-v7.1.tgz && cd cuda
sudo cp -P cuda/include/cudnn.h /usr/include
sudo cp -P cuda/lib64/libcudnn* /usr/lib/x86_64-linux-gnu/
sudo chmod a+r /usr/lib/x86_64-linux-gnu/libcudnn*