Ubuntu16.04安装CUDA8.0+CUDNN6
本机已安装nvidia显卡驱动,再此基础上安装cuda+cudnn
1.安装依赖项
sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
cuda:
https://developer.nvidia.com/cuda-80-ga2-download-archive
https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu
cudnn: https://developer.nvidia.com/rdp/cudnn-archive
2.安装
sudo chmod +x cuda_8.0.44_linux.run
sudo ./cuda_8.0.44_linux.run
步骤见下:
更多,按q键
accept
n
y
回车
y
y
回车
3、环境配置
sudo gedit ~/.bashrc
//文件末尾添加
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
source ~/.bashrc
4.验证
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make
sudo ./deviceQuery
5.CUDNN6 安装
tar -zxvf cudnn-8.0-linux-x64-v6.0.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.6
sudo ln -s libcudnn.so.6.0.21 libcudnn.so.6
sudo ln -s libcudnn.so.6 libcudnn.so
sudo ldconfig
6.查看cudnn版本号
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
7.查看cuda版本号
nvcc --version
参考博客:https://blog.csdn.net/qq_40806289/article/details/90643603