Ubuntu 14.04下CUDA8.0 + cuDNN v5 + Caffe 安装配置(2)

等待编译完成即可。
注意:编译需要非常大的内存和非常长的时间。一般情况下不会报错。报错查阅相关书籍和搜索引擎,此处不关注。

(4) 配置 caffe 环境

caffe 运行时需要调用 cuda 的库,我们在/etc/ld.so.conf.d目录下新建一个cafe.conf文件。终端输入
$ sudo gedit /etc/ld.so.conf.d/caffe.conf
打开 gedit 。添加内容:
/usr/local/cuda/lib64
保存退出。
更新配置
$ sudo ldconfig

所有配置结束。

三、测试

此处说的很简略。假设你已经初步掌握 caffe 的用法。

# cd ~/caffe # sudo sh data/mnist/get_mnist.sh # sudo sh examples/mnist/create_mnist.sh # sudo sh examples/mnist/train_lenet.sh

感受1080开启cuDNN模式的强大吧!

Ubuntu 14.04下CUDA8.0 + cuDNN v5 + Caffe  安装配置


caffe 例程结果

一切搞定。

Ubuntu 14.04 安装配置CUDA 

Ubuntu 12.04配置NVIDIA CUDA5.5实录 

Ubuntu安装Theano+CUDA 

关于Ubuntu 12.04 下 CUDA5.5 的安装请参看如下链接 Ubuntu 12.04 安装 CUDA-5.5

Caffe配置简明教程 ( Ubuntu 14.04 / CUDA 7.5 / cuDNN 5.1 / OpenCV 3.1 )

在Ubuntu 14.04上配置CUDA+Caffe+cuDNN+Anaconda+DIGITS 

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:https://www.heiqu.com/14498.html