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 版本問題,參考 解壓,編譯...