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.

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值