本文以在Ubuntu 1604 環境下,CUDA 10.0 CUDNN 7.6.0 爲例子介紹如何檢測CUDNN是否成功安裝。
cuDNN 7.6.0 安裝:
tar -xvf cudnn-10.0-linux-x64-v7.tgz
cd cuda
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.7
sudo ln -s libcudnn.so.7.6.0 libcudnn.so.7
sudo ln -s libcudnn.so.7 libcudnn.so
sudo ldconfig
cuDNN 7.6.0 檢測是否成功安裝:
下載鏈接:https://developer.nvidia.com/rdp/cudnn-archive
下載三個文件:cuDNN v7.6.0 Runtime Library for Ubuntu16.04 (Deb)
cuDNN v7.6.0 Develpoper Library for Ubuntu16.04 (Deb)
cuDNN v7.6.0 Code Samples and User Guide for Ubuntu16.04 (Deb)
sudo dpkg -i libcudnn7_7.6.0.64-1+cuda10.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.6.0.64-1+cuda10.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.6.0.64-1+cuda10.0_amd64.deb
cp -r /usr/src/cudnn_samples_v7/ /home/cudnntest
cd /home/cudnntest/cudnn_samples_v7/mnistCUDNN
make make
./mnistCUDNN
如果成功應該顯示 Test passed !