CUDA

cuDNN 安装

2017-03-01  本文已影响350人  盲狙小堡垒
前提

已经安装好了CUDA,并且知道cuda目录(一般是/usr/local/cuda)
如果不知道可以用以下命令查询

which nvcc

准备文件

下载你需要的版本:https://developer.nvidia.com/rdp/cudnn-download

例如我下载cnDNN5(基于CUDA 8.0 和 Linux)(此操作不一定要在cuda安装目录下,URL地址需要根据你的版本来替换)

wget https://developer.nvidia.com/rdp/assets/cudnn-7.5-linux-x64-v5.0-ga-tgz

tar -xvf cudnn-7.5-linux-x64-v5.0-ga-tgz

安装(复制文件+创建软连接)

其实cuDNN的安装就是把几个文件复制到which nvcc显示的目录下。把解压好后的cuda里面的include 和 lib64 都复制到你的CUDA目录下(一般是/usr/local/cuda)
然后重建一下软连接。

ln -s libcudnn.so.5.1.3 libcudnn.so.5.1
ln -s libcudnn.so.5.1 libcudnn.so

OK, now We are done!

上一篇下一篇

猜你喜欢

热点阅读