之前对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里,你想考到哪里都行。
网友评论