ubuntu16.04安装cuda+cudnn

1 安装cuda

Do you accept the previously read EULA?
accept/decline/quit: accept

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 410.48?
(y)es/(n)o/(q)uit: n

Install the CUDA 10.0 Toolkit?
(y)es/(n)o/(q)uit: y

Enter Toolkit Location
 [ default is /usr/local/cuda-10.0 ]: 回车

Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: y

Install the CUDA 10.0 Samples?
(y)es/(n)o/(q)uit: y

Enter CUDA Samples Location
 [ default is /home/linkdome ]: 回车

11.x 版本的安装界面是:
在这里插入图片描述最后选择install即可。

添加环境变量

sudo gedit ~/.bashrc

添加语句在文档末尾:

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

更新源:

source ~/.bashrc

验证是否安装成功:
输入:nvcc -V,显示下图内容即为成功。
在这里插入图片描述

2 安装cudnn

对下载的压缩包解压:

tar -xvf cudnn-10.2-linux-x64-v7.6.5.32.tgz

解压过程如下:

cuda/include/cudnn.h
cuda/NVIDIA_SLA_cuDNN_Support.txt
cuda/lib64/libcudnn.so
cuda/lib64/libcudnn.so.7
cuda/lib64/libcudnn.so.7.6.5
cuda/lib64/libcudnn_static.a

执行以下命令:

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

验证是否安装成功:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
#当cudnn版本过高时,可使用如下语句
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

输出结果为下图所示,即为成功:
在这里插入图片描述


版权声明:本文为wuhongwuyan原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。