測試CUDNN是否成功安裝

本文以在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 !

 

 

 

 

 

 

 

 

 


 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章