ubuntu下,cuDNN的安装

一、下载

到下面网址下载(需要登陆账号)

Log in | NVIDIA Developer

二、解压

对下载的cudnn-xxxxxxxxxxxxx.tgz进行解压操作,得到一个文件夹cuda,命令为:

tar -zxvf xxxxxxxxxxxxx.tgz

然后,使用下面两条指令复制cuda文件夹下的文件到/usr/local/cuda/lib64/和/usr/local/cuda/include/中。

sudo cp cuda/lib/* /usr/local/cuda/lib64/
sudo cp cuda/include/* /usr/local/cuda/include/

注意,local文件夹下的cuda是个快捷方式,如下图

拷贝完成后,我们可以使用如下的命令查看cuDNN的信息:

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

 安装成功!

猜你喜欢

转载自blog.csdn.net/gaoenyang760525/article/details/131256427