ubuntu 16.04中CAFFE配置步骤

显卡驱动

sudo apt-get purge nvidia-*
sudo apt-get update
sudo apt-get purge nvidia-*
sudo apt-get-rep
sudo apt-get-repository ppa:graphics-
ifconfig
sudo apt-get
sudo apt-get install openssh-server
service ssh start
sudo add-apt-repository ppa:graphics-drivers/ppa and then sudo apt-get update
sudo apt-get update
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update && sudo apt install nvidia-364
sudo reboot
sudo rm -fr ~/.cache/compizconfig-1
sudo rm -fr ~/.compiz
sudo rm -fr ~/.Xauthority 
sudo rm -fr ~/.config/autostart
sudo apt-get install --reinstall ubuntu-desktop unity compizconfig-settings-manager upstart
sudo dconf reset -f /org/compiz/
setsid unity
startx
sudo apt-get install lightdm
startx

配置gnome桌面

sudo apt-get install gnome-session-flashback
sudo apt-get install compizconfig-settings-manager
# gnome-tweak-tool

配置CUDA和CUDANN

下载CUDA本地安装,注意是否覆盖安装NVIDIA驱动。安装完CUDA之后,一定要成功编译CUDA Samples和CUDANN的例子。

配置Caffe

  • leveldb
git clone ...
sudo cp --preserve=links libleveldb.* /usr/local/lib
sudo cp -r include/leveldb /usr/local/include

使用GCC5强行编译

cmake ... -DCUDA_NVCC_FLAGS="-D_FORCE_INLINES" ..

配置MATLAB

根据matconvnet配置,这个最简单。若需要matcaffe,则需要make matcaffe。

gpuDevice()
run matlab/vl_setupnn
addpath('CAFFE-ROOT/matlab/+caffe/private')

后记和总结

在配置过程中与到了很多问题:

  • GCC版本问题,Ubuntu 16.04默认的是GCC5.3,但会在编译CUDA中出现string.h memcpy函数的问题,我的做法是不对GCC降级,强行使用GCC5编译。

  • 安装caffe 需要CUDA和CUDANN,注意修改配置文件。

  • make caffe时候,注意出现的错误,安装。不要随便rm系统文件,非常危险。最好记住安装的大致进度,可能安装一个库会引起其他错误。

Link

Ubuntu 14.04 + Caffe + Cuda 7.5 + Opencv 3.0安装教程
NVIDIA CUDA with Ubuntu 16.04 beta on a laptop (if you just cannot wait)
Caffe + Ubuntu 15.04 + CUDA 7.0 新手安装配置指南
从零安装 Caffe (Ubuntu 14.04)
Ubuntu 下配置protobuf
Ubuntu 16.04 or 15.10 OpenCV 3.1 Installation Guide

推荐阅读更多精彩内容