ubuntu16.04查看cudnn,cuda的版本,同時也可以檢驗cuda,cudnn是否真的安裝成功
首先,我們知道NVIDIA官網更新了所有的cudnn安裝包,之前都是tar.gz格式,我們是通過解壓文件在安裝。但是現在如果是通過Deb包安裝,則查看版本是不一樣的,檢驗是否安裝成功命令也不太一樣。
1.如果是tar.gz格式的安裝
查看cuda的版本:
cat /usr/local/cuda/version.txt
查看cudnn 版本:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
2.如果是Deb安裝版本
比如我們安裝的是7.0.5版本cudnn,則要下載三個文件:
cuDNN v7.0.5 Runtime Library for Ubuntu16.04 (Deb)
cuDNN v7.0.5 Develpoper Library for Ubuntu16.04 (Deb)
cuDNN v7.0.5 Code Samples and User Guide for Ubuntu16.04 (Deb)
cudnn下載
此時,
查看cuda的版本:
cat /usr/local/cuda/version.txt
查看cudnn 版本:
使用原來的cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
命令會提示找不到cudnn.h文件,原因是
新版cudnn Deb安裝版本 找不到cudnn.h,他不會有這個文件。
應該使用命令:
dpkg -l | grep cudnn
比如我的是cudnn7,有三個文件。
檢驗cudnn是否安裝成功
找到cudnn_samplesv7文件夾,
如圖,我的檢驗方法是:
cd mnistCUDNN
運行裏面的mnistCUDNN腳本
./mnistCUDNN
顯示出 Test passed!說明安裝成功。