美文网首页
caffe 介绍和编译

caffe 介绍和编译

作者: 张扬_c572 | 来源:发表于2018-10-20 00:22 被阅读0次

之前对caffe也不是完全不了解,就这玩意儿是个在读博士做的(人和人。。。。)。

https://github.com/BVLC/caffe.git  

编译这个玩意儿可不是个简单的活儿,对于一些没咋编译过大型软件的童鞋来讲。

1驱动安装

需要一张n卡,禁用ubuntu自带驱动,然后安装,重启。我1080ti,ubuntu16.04lts, nvidia-384驱动。

sudo touch /etc/modprobe.d/blacklist.conf

sudo echo 'blacklist nouveau' >> /etc/modprobe.d/blacklist.conf

sudo echo 'options nouveau modeset=0' >> /etc/modprobe.d/blacklist.conf

sudo add-apt-repository ppa:graphics-drivers/ppa

sudo apt-get update && sudo apt-get install nvidia-384

sudo reboot

2CUDA安装

 CUDA™是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。说白了就是你是你平时用CPU算东西很自然,但是你现在想用GPU算东西,就得用这个框架。神经网络是一个天然需要并行计算的完美任务,所以自然需要CUDA。

去官网下个和驱动适配的.run安装包,运行即可。

安装在/usr/local/cuda-x.x中,还有个/usr/local/cuda的软连接。

3CUDNN

针对英伟达显卡的NN(神经网络)开发包,自然需要,查一下和CUDA的适配。

解压缩然后

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d

sudo chmod a+r /usr/local/cuda/include/cudnn.h

sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

4caffe

我个人觉得上面几步比较烦,下面对于熟悉c++的人来说还好。

首先安装一大堆依赖,按照官网的来就好了

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler

sudo apt-get install --no-install-recommends libboost-all-dev

sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

修改一哈makefile,主要就是hdf5_serial的问题。

修改一哈makefile.config 修改你的各种blas,cuda,还有那些所有以来位置。

然后make一哈。

然后在make pycaffe一哈。

声明在caffe/include里,库文件在caffe/build/lib里,你想考到哪里都行。

相关文章

网友评论

      本文标题:caffe 介绍和编译

      本文链接:https://www.haomeiwen.com/subject/hlkpzftx.html