1. 程式人生 > >ubuntu16.04安裝cudnn(nvidia 384,cuda9.0)

ubuntu16.04安裝cudnn(nvidia 384,cuda9.0)

問題:在ubuntu16.04上安裝cudnn,本機已經提前安裝NVIDIA384和cuda9.0

解除安裝之前安裝的cudnn:

sudo rm -rf /usr/local/cuda/include/cudnn.h

sudo rm -rf /usr/local/cuda/lib64/libcudnn*

解決:

Linux安裝有兩種方式,一種下載tar包,一種下載deb包安裝,這裡推薦下載deb包安裝,便於後面進行是否安裝成功的驗證。

來到對應的下載目錄,按照下面的命令進行安裝:

sudo dpkg -i libcudnn7_7.3.1.20-1+cuda9.0_amd64.deb

sudo dpkg -i libcudnn7-dev_7.3.1.20-1+cuda9.0_amd64.deb

sudo dpkg -i libcudnn7-doc_7.3.1.20-1+cuda9.0_amd64.deb

執行完畢就安裝成功了。

是否安裝成功驗證:

cp -r /usr/src/cudnn_samples_v7/ /home/

cd /home/cudnn_samples_v7/mnistCUDNN

make clean && make

./mnistCUDNN

輸出的最後如下即表示安裝成功