史上最完整的caffe环境搭建教程

  • Post author:
  • Post category:其他


从一个干净的新系统开始搭建 caffe

硬件:GPU GTX 1060 6G

OS:ubuntu16.04 x64

安装系统后,重启后,如果在启动界面出现


fb switching to nouveaufb from efi vga


这个错误,则拔掉显卡, 然后安装显卡驱动,再插上显卡就可以了

安装显卡驱动:

通过

这里

找到驱动版本(我现在用的版本是384),然后通过这个命令安装驱动


sudo apt-get install nvidia-384


如果有错误,可以先执行


sudo apt-get upgrade


之后再执行安装驱动的命令

CUDA:


这里下载


安装


sudo sh ./cuda_8.0.61_375.26_linux.run


然后等一会儿会让你读


End User License Agreement


直接按CTRL + C吧

按照提示去安装吧,有点要注意下,提示再次安装驱动的时候要选NO,因为我们已经安装过了

安装成功之后

sudo vi /etc/ld.so.conf
#最后面添加这行
/usr/local/cuda/lib64/
#保存并执行
sudo ldconfig

安装各种工具和库

sudo apt install cmake
sudo apt install unzip
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install  libleveldb-dev
sudo apt-get install libsnappy-dev
sudo apt-get install libatlas-base-dev
sudo apt-get install libhdf5-serial-dev
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt install protobuf-compiler
sudo apt-get install libprotobuf-dev
sudo apt install python-pip

编译安装opencv3

wget https://codeload.github.com/opencv/opencv/zip/3.4.0
unzip opencv-3.4.0.zip
cd unzip opencv-3.4.0.zip
mkdir x64
cd x64
cmake ../  #ippicv_2017u3_lnx_intel64_general_20170822 由于网络原因,下载这个文件可能会失败, 不过不影响
make
sudo make install

有GPU的话, 编译好漫长, 要编译几个小时

caffe:

安装和python相关的库,iPython可能会安装失败,具体干嘛的我也不清楚,无视

git clone https://github.com/BVLC/caffe.git
cd caffe/python/
for req in $(cat requirements.txt); do sudo pip install $req; done

进到caffe的源码目录

cp Makefile.config.example Makefile.config
vi Makefile.config
#除掉 OPENCV_VERSION := 3 之前的#号
#在 INCLUDE_DIRS := 最后面加上 /usr/include/hdf5/serial  
#在LIBRARY_DIRS 最后面加上 /usr/lib/x86_64-linux-gnu/hdf5/serial
#保存
make pycaffe
make all
make test
make runtest

如果在编译的时候出现

E0104 10:41:13.060823  3649 common.cpp:114] Cannot create Cublas handle. Cublas won't be available.
E0104 10:41:13.064980  3649 common.cpp:121] Cannot create Curand generator. Curand won't be available.
F0104 10:41:13.065028  3649 syncedmem.hpp:22] Check failed: error == cudaSuccess (30 vs. 0)  unknown error

则执行

sudo ldconfig /usr/local/cuda/lib64

最后

vi ~/.bashrc
#最后面加上 export PYTHONPATH=/home/dyjun/deeplearn/caffe/python/
#保存并退出
source ~/.bashrc

大功告成~



版权声明:本文为diaoying原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。