ubuntu 16.04中CAFFE配置步驟

weixin_34146805發表於2016-06-25

顯示卡驅動

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

相關文章