TX1刷機(jī),配置CUDA及Caffe安裝

一、環(huán)境準(zhǔn)備

用JetPack刷機(jī)到16.04LTS
安裝Cuda 8.0.34和CuDnn v5.1

二、安裝caffe

2.1依賴庫安裝

sudo add-apt-repository universe
sudo apt-get update
sudo apt-get install cmake git aptitude screen g++ libboost-all-dev \
libgflags-dev libgoogle-glog-dev protobuf-compiler libprotobuf-dev \
bc libblas-dev libatlas-dev libhdf5-dev libleveldb-dev liblmdb-dev \
libsnappy-dev libatlas-base-dev python-numpy libgflags-dev \
libgoogle-glog-dev python-skimage python-protobuf python-pandas \
libopencv-dev

opencv也可以根據(jù)自己的需要選擇版本編譯,這里為了方便就這么安裝了

2.2 下載caffe

直接從BCLV的Git中下載

git clone https://github.com/BVLC/caffe.git 

不用使用Nvidia提供的caffe(32Bit),TX1是64Bit的

2.3編譯caffe

修改caffe的配置文件

cd caffe
cp Makefile.config.sample Makefile.config
vim Makefile.config
去掉注釋,使用CuDNN,使用python層

然后編譯
make all -j4
用自帶的mnist測試一下

$ bash ./date/mnist/get_mnist.sh  
$ bash ./examples/mnist/create_mnist.sh  
$ bash ./examples/mnist/train_lenet.sh 

有個(gè)坑爹的情況是嚴(yán)樂春大哥的網(wǎng)站可能會(huì)崩,所以數(shù)據(jù)下不全,自己拷一下數(shù)據(jù)就好。

2.4使用g++編譯caffe中的classification.cpp文件

參考博客[1]

2.5編譯pycaffe

講道理是沒什么問題的,但是忘記編譯和測試了,下次搞了再更新。

參考博客

http://blog.csdn.net/q6324266/article/details/52193076
http://blog.csdn.net/zouyu1746430162/article/details/53391283

Trouble shooting

需要注意的是HDF5的頭文件路徑和庫路徑

HDF5_INCLUDE_DIRS := /usr/include/hdf5/serial/
HDF5_LIB_DIRS := /usr/lib/aarch64-linux-gnu/hdf5/serial
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include $(HDF5_INCLUDE_DIRS)
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib $(HDF5_LIB_DIRS)
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容