Caffe2的安裝詳述

2021-08-11 04:06:39 字數 2259 閱讀 8999

caffe2的安裝過程區分ubuntu的不同版本,在ubuntu 14.04和 ubuntu 16.04下的相關命令有所不同,最重要的是caffe2主要支援python2.7版本,以下python和pip命令預設情況下都是指python2.7版本,經過筆者測試,caffe2版本暫時不支援python3.5及以上版本。

1、首先,安裝required dependencies依賴包 命令列如下:

sudo apt-get update

sudo apt-get install -y --no-install-recommends \

build-essential \

cmake \

git \

libgoogle-glog-dev \

libprotobuf-dev \

protobuf-compiler \

python-dev \

python-pip                          

sudo pip install numpy protobuf

2、其次,如果有gpu支援的話optional gpu support,需要安裝cuda型別的依賴包,具體分為 ubuntu 14.04和 ubuntu 16.04,具體命令如下:

注意區分ubuntu的不同版本:

2.1 for ubuntu 14.04

sudo apt-get update && sudo apt-get install wget -y --no-install-recommends

wget ""

sudo dpkg -i cuda-repo-ubuntu1404_8.0.61-1_amd64.deb

sudo apt-get update

sudo apt-get install cuda

2.2 for ubuntu 16.04

sudo apt-get update && sudo apt-get install wget -y --no-install-recommends

wget ""

sudo dpkg -i cuda-repo-ubuntu1604_8.0.61-1_amd64.deb

sudo apt-get update

sudo apt-get install cuda

2.3 install cudnn (對於所有ubuntu版本)

cudnn_url=""

wget $

sudo tar -xzf cudnn-8.0-linux-x64-v5.1.tgz -c /usr/local

rm cudnn-8.0-linux-x64-v5.1.tgz && sudo ldconfig

3、其次,如果沒有gpu支援,只考慮cpu的支援,繼續安裝其他依賴包

注意:libgflags2包是為ubuntu 14.04準備的,而libgflags-dev包是為ubuntu 16.04準備的。

3.1 for ubuntu 14.04

sudo apt-get install -y --no-install-recommends libgflags2

3.2 for ubuntu 16.04

sudo apt-get install -y --no-install-recommends libgflags-dev

3.3 for both ubuntu 14.04 and 16.04 (對於所有ubuntu版本)

4、然後,開始複製和構建caffe2,命令列如下:

git clone --recursive && cd caffe2

sudo make && cd build && sudo make install

sudo python -c 'from caffe2.python import core' 2>/dev/null && echo "success" || echo "failure"

5、最後,如果第4步中最後一行命令返回」success「,則表明caffe2安裝成功。

6、如果之前成功安裝了gpu支援依賴包,可以執行如下命令列來測試,但注意到如果沒有gpu包,則caffe2會選擇cpu來代替,但這一代替會出現類似丟失庫檔案的錯誤。

python -m caffe2.python.operator_test.relu_op_test

caffe2 安裝與介紹

一早發現caffe2的較成熟的release版發布了 the first production ready release 那麼深度學習平台在之後一段時間也是會出現其與tensorflow相互競爭的局面。從開啟這個caffe2的官網就會發現,有了facebook的支援,連介面也好看多了。不過再仔細看...

樹莓派編譯安裝caffe2

平心而論,樹莓派平台可以選擇的深度學習框架還是相當多的,這裡主要記錄下caffe2平台的安裝。首先按照官網教程,clone原始碼 git clone recursive git submodule update init考慮到樹莓派可能不是很穩定,用ubuntu系統clone下來打包丟到樹莓派裡邊也...

caffe 轉caffe2模型出現的問題

剛開始接觸caffe,用nvidia tx2跑了ssd模型,達到8 9幀。把caffe模型轉caffe2模型時,轉到reshape層時出現錯誤 valueerror unknown argument type key values value 發現是protobuffer 版本問題,參考 解壓,編譯...