1. 程式人生 > >GTX 1080Ti + cuda8.0 + cuDNN6.0 安裝及測試

GTX 1080Ti + cuda8.0 + cuDNN6.0 安裝及測試

GPU 顯示卡廠商已經安裝好了,直接安裝 cuda8.0 + cuDNN6.0
我這裡的顯示卡是 GTX 1080 Ti

cuda安裝
我下載的是cuda8.0的是deb格式的1.9個G地址:https://developer.nvidia.com/cuda-downloads
cd到下載目錄
sudo dpkg -i cuda-repo-ubuntu1404-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda

設定環境變數

\ sudo gedit /etc/profile
開啟/etc/profie檔案,在檔案末尾新增如下設定:
export PATH=/usr/local/cuda-8.0/bin:$PATH
儲存該檔案,執行以下命令,使環境變數立即生效。
\$ source /etc/profile

 新增lib路徑
$ sudo gedit /etc/ld.so.conf.d/cuda.conf
在/etc/ld.so.conf.d中新增檔案cuda.conf,該檔案中寫入如下內容:
/usr/local/cuda/lib64
儲存該檔案,執行下列命令使其立即生效。
$ sudo ldconfig

測試 cuda 安裝是否成功

cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery

sudo make

./deviceQuery

//如果顯示pass,則代表成功。

安裝cuDNN比較簡單 先下載 cudnn-8.0-linux-x64-v6.0.tgz
解壓後把相應的檔案拷貝到對應的CUDA目錄下即可:

滑鼠右鍵 extract here 得到 cuda 資料夾

顯示以下資訊:
*cuda/include/cudnn.h
cuda/lib64/libcudnn.so
cuda/lib64/libcudnn.so.6
cuda/lib64/libcudnn.so.6.0.21
cuda/lib64/libcudnn_static.a*

繼續執行以下指令:
cd 到下載目錄 Downloads

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

GTX1080 LetNet-5 CPU GPU cuDNN6.0 時間對比

CPU模式:
Makefile.config

CPU_ONLY:=1 前取消 #
make clean
make -j

./build/tools/caffe.bin time -model examples/mnist/lenet_train_test.prototxt
Testing for 50 iterations

Total Time: 2961 ms.

GPU 模式:

Makefile.config

#CPU_ONLY:=1
make clean
make -j

./build/tools/caffe.bin time -model examples/mnist/lenet_train_test.prototxt -gpu 0

Total Time: 501.244 ms.

cuDNN 模式:

Makefile.config

USE_CUDNN:=1

make clean
make -j

./build/tools/caffe.bin time -model examples/mnist/lenet_train_test.prototxt -gpu 0

Testing for 50 iterations.
Total Time: 73.0883 ms.