平心而論,樹莓派平台可以選擇的深度學習框架還是相當多的,這裡主要記錄下caffe2平台的安裝。
首先按照官網教程,clone原始碼:
git clone --recursive
git submodule update --init
考慮到樹莓派可能不是很穩定,用ubuntu系統clone下來打包丟到樹莓派裡邊也許會好些。。。
之後由於樹莓派記憶體限制,我們要先把樹莓派的交換區擴大到2g。輸入:
sudo nano /etc/dphys-swapfile
將conf_swapsize=100改為=2048以上,儲存退出。之後用下列命令啟用交換區:
sudo /etc/init.d/dphys-swapfile stop
sudo /etc/init.d/dphys-swapfile start
之後由於caffe2合併到pytorch了,所以應該用下列指令編譯,而不是如官網所述cd caffe2:
cd pytorch
./scripts/build_raspbian.sh
之後根據報錯的提示,pip安裝相應軟體包,我這邊安裝了pyyaml,不過我的樹莓派在之前安裝了tengine平台,不清楚會不會附帶安裝了一些依賴外掛程式,等不報錯開始正常編譯之後就是等了,大概3~4小時吧,編譯完成之後,就可以吧交換區改回100了,畢竟設定的太大對tf卡不好。
編譯完成之後安裝:
cd build
sudo make install
現在開啟python(注意不是python3),輸入import caffe2,不報錯說明這一步安裝成功。
以下操作出自
輸入import caffe2.python.predictor.predictor_exporter as pe,正常情況下會報錯,因為python中protobuf的版本和系統版本不一致,可通過下列指令檢視:
pkg-config --modversion protobuf
pip list | grep protobuf
不一致沒關係,我們執行
sudo pip install --upgrade protobuf==3.6.1 #根據你系統的版本改
沒安裝future庫的要裝上,不然報錯
sudo pip install future
更新protobuf失敗可能是pip版本過低,需要執行:
pip install --upgrade pip
sudo pip install --upgrade pip
好啦,現在caffe2就編譯安裝完畢了。 caffe2 安裝與介紹
一早發現caffe2的較成熟的release版發布了 the first production ready release 那麼深度學習平台在之後一段時間也是會出現其與tensorflow相互競爭的局面。從開啟這個caffe2的官網就會發現,有了facebook的支援,連介面也好看多了。不過再仔細看...
Caffe2的安裝詳述
caffe2的安裝過程區分ubuntu的不同版本,在ubuntu 14.04和 ubuntu 16.04下的相關命令有所不同,最重要的是caffe2主要支援python2.7版本,以下python和pip命令預設情況下都是指python2.7版本,經過筆者測試,caffe2版本暫時不支援python...
Caffe2 新增CUDA路徑
在使用eclipse時候,仍然出現 libcuda.so 找不到的情況.原因 cuda環境沒有設定到系統環境中。方案 設定cuda path 在bashrc中新增路徑,只對當前使用者產生影響。sudo gedit bashrc在最後行新增內容 export path usr local cuda b...