前提
已經(jīng)安裝好了CUDA,并且知道cuda目錄(一般是/usr/local/cuda)
如果不知道可以用以下命令查詢
which nvcc
準(zhǔn)備文件
下載你需要的版本:https://developer.nvidia.com/rdp/cudnn-download
例如我下載cnDNN5(基于CUDA 8.0 和 Linux)(此操作不一定要在cuda安裝目錄下哪审,URL地址需要根據(jù)你的版本來替換)
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
安裝(復(fù)制文件+創(chuàng)建軟連接)
其實cuDNN的安裝就是把幾個文件復(fù)制到which nvcc顯示的目錄下市框。把解壓好后的cuda里面的include 和 lib64 都復(fù)制到你的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!