GTX 1080Ti + cuda8.0 + cuDNN6.0 安装及测试
GTX 1080Ti + cuda8.0 + cuDNN6.0 安装及测试
版权声明：本文为博主原创文章，未经博主允许不得转载。 /developer/article/1437720
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.