由於在實習工作中需要在tk1環境下安裝caffe,碰到很多坑,因此記錄下來,一來希望幫助別人少走彎路,二來做個記錄避免自己忘記
1.打開終端輸入如下
sudo add-apt-repository universe sudo apt-get update sudo apt-get install libprotobuf-dev protobuf-compiler gfortran \ libboost-dev cmake libleveldb-dev libsnappy-dev \ libboost-thread-dev libboost-system-dev \ libatlas-base-dev libhdf5-serial-dev libgflags-dev \ libgoogle-glog-dev liblmdb-dev gcc-4.7 g++-4.7結果如下
2.下載 caffe點擊打開鏈接,可以直接點擊圖中的Dowload ZIP,此時文件名爲caffe-master,當然也可以用git clone,下載下來的文件名爲caffe,將文件移動到home
3.下載和解壓cuda
4.安裝cuda-toolkit
sudo apt-get update
sudo apt-get install cuda-toolkit-6-5 -y
5.修改環境變量
$user is your username
sudo usermod -a -G video $USER
echo "# Add CUDA bin & library paths:" >> ~/.bashrc
echo "export PATH=/usr/local/cuda-6.5/bin:$PATH" >> ~/.bashrc
echo "export LD_LIBRARY_PATH=/usr/local/cuda-6.5/lib:$LD_LIBRARY_PATH" >> ~/.bashrc
source ~/.bashrc
6.運行cuda的sample
cd /usr/local/cuda-6.5/bin
./cuda-install-samples-6.5.sh ~/
時間略長...
/usr/local/cuda
nvcc -V
8.查看
cd ~/NVIDIA_CUDA-6.5_Samples/bin/armv7l/linux/release/gnueabihf
ls
安裝成功
9.修改Makefile
cd caffe-master
cp Makefile.config.example Makefile.config
vim Makefile.config
如果使用opencv3,
修改使得下面命令生效
OPENCV_VERSION := 3
由於我們使用cuda6.5,comment掉下圖的部分
10.執行命令
sudo make -j4
出現報錯信息:Error:cannot find lboost_filesystem ,解決方法爲輸入如下命令
sudo apt-get install libboost-all-dev
sudo make -j4
11.運行test example,結果如下圖所示
至此說明安裝成功。