樹莓派編譯安裝caffe2

2021-09-19 05:52:02 字數 1413 閱讀 4180

平心而論,樹莓派平台可以選擇的深度學習框架還是相當多的,這裡主要記錄下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...