从一个干净的新系统开始搭建 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
大功告成~