天天看點

tx2安裝caffe總結

無奈筆記本的性能太渣,雙系統切換太麻煩,索性就拿tx2來當第二台電腦,需要在linux上跑的demo都放到tx2上跑;

先安裝caffe(我重刷了兩次機o(『﹏『)o)。

先配置依賴項

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler

sudo apt-get install –no-install-recommends libboost-all-dev

(看他人的部落格要安裝libopencv-dev,但是我安裝時出現opencv版本的問題,是以沒安裝,在使用opencv的python庫時并沒有出現什麼問題,是以暫且不管)

然後是python的相關安裝:

sudo apt-get install python-dev

sudo apt-get install python-numpy

sudo apt-get install ipython

sudo apt-get install ipython-notebook

sudo apt-get install python-sklearn

sudo apt-get install python-skimage

sudo apt-get install python-protobuf

當然還要安裝pip

sudo apt-get install python-pip

然後sudo pip install –upgrade pip

但是用pip安裝python包時會出現速度非常慢的現象,安裝一個包少則四五分鐘,多則十幾分鐘,可能是tx2的cpu核心是A57,和X86比還是有很大差距,導緻編譯時慢(看top猜的),是以建議盡量用apt-get安裝

然後是谷歌glog和gflags和lmdb依賴項

sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

接着安裝git,并且下載下傳代碼

sudo apt-get install git

進入到源碼

cd caffe

cp Makefile.config.example Makefile.config

修改makefile和Makefile.config

vim Makefile.config

去掉USE_CUDNN := 1前的注釋,然後去掉

去掉89行左右的WITH_PYTHON_LAYER := 1前的注釋

為了比對cuda8的計算能了去掉

-gencode arch=compute_20,code=sm_20 \

-gencode arch=compute_25,code=sm_25 \

在92行左右将

改為

然後改Makefile.config

将175行左右的

然後

sudo make clean

sudo make -j8

tx2安裝caffe總結

在這裡caffe的安裝已經完成,來用一下caffe自帶的minst demo來體驗一下:

cd caffe;

sudo sh data/minst/get_mnist.sh

下載下傳mnist的資料集

sudo sh example/mnist/create_mnist.sh

sudo sh example/mnist/train_lenet.sh

tx2安裝caffe總結

可以看到caffe識别出TX2的GPU并調用TX2的GPU在跑demo

tx2安裝caffe總結

從結果看caffe的mnist sample用了兩層卷積,跑完訓練用了2分多鐘,并且準确率達到99%,(比matlab跑快好多;雖然網上說GPU+CUDNN隻要45秒,但這邊要兩分鐘,可能是TX2畢竟屬于移動版,還是比不上GTX系列或者專業顯示卡)(注意caffe的example的sample檔案裡是相對路徑的相對于caffe檔案夾,是以cd到caffe檔案加下運作sh)

這邊我再說一下我踩得兩個坑:

1:在跑比如github上的rcnn時,編譯github會出現Makefile 563的錯誤,那是因為github上caffe-rcnn包中的cudnn版本的原因,解決方法是将下載下傳的caffe中src/layers/,src/util/,include/caffe/layers/,include/caffe/util/檔案中的cudnn檔案複制替換caffe-rcnn檔案的相應位置的cudnn檔案

2:在運作如rcnn的demo.py時會出現類似

那時因為在nms檔案夾下的檔案是對于桌面平台的GPU及CPU編譯的,你需要删除然後進入lib目錄下

執行make指令,編譯時可能會出現相應的python包沒安裝,隻要pip或apt-get就行了

然後再運作demo.py就可以了

tx2安裝caffe總結

繼續閱讀