1. 程式人生 > >Ubuntu下快速配置Caffe

Ubuntu下快速配置Caffe

Caffe安裝

實際上在windows上安裝過多次caffe了,無論是BLVC版本的還是Microsoft版本的,ubuntu的按照也進行過,這段時間在自己筆記本上
又折騰了下caffe安裝,發現其實直接照著官方的是最方便快捷的。
具體可以參照 Installation instructions

具體步驟

根據系統的不同,ubuntu分為兩種安裝方式。Ubuntu17.04以即上的可以直接apt-get

  sudo apt install caffe-cpu # cpu only
  sudo apt install caffe-cuda # gpu

其他版本的ubuntu也可以分為兩種方式安裝,但是依賴項是必須的,本文僅介紹簡易的軟體源中快速安裝。原始碼安裝可以參考本人的另外一篇部落格:

Nvidia顯示卡驅動

由於要使用GPU,所以先要檢視自己顯示卡所匹配的顯示卡驅動,網址:nvidia ,下載run檔案。
由於目前顯示卡和cuda更新迅速,容易造成筆記本迴圈登入,因此安裝顯示卡驅動是關閉圖形介面。

  # ctrl +alt +F1 進入tty1,
  sudo service lightdm stop
  sudo ./Nvidia-.....run 執行安裝
  sudo reboot

Cuda和CuDnn

安裝較為簡單,官網下載,在安裝cuda是需要注意顯示卡安裝選項選擇no即可。

    sudo sh cuda_8.0.44_linux.run --override
    # 安裝結束後
    sudo vim ~/.bashrc  //末尾新增
    export CUDA_HOME=/usr/local/cuda-8.0
    export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
    export PATH=/usr/local/cuda-8.0/bin:$PATH
    source ~/.bashrc

    # 測試
    cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
    make -j32
    sudo ./deviceQuery

cudnn下載後接下的include lib拷貝到cuda的安裝路徑,並設定連結。

   sudo ln -s libcudnn.so.xxx libcudnn
   sudo ln -s libcudnn.so.xx libcudd.so
   sudo ldconfig

其他依賴項

其他依賴項安裝可以直接從軟體源獲取,當然也可以自己原始碼安裝。

  # protobuf,snappy,leveldb,opencv,hdf5,boost ,python-opencv,glog ,gflag,lmdb
  sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
  sudo apt-get install --no-install-recommends libboost-all-dev
  sudo apt-get install python-dev python-opencv
  sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

關於blas可以選擇atlas,openblas和MKL,由於後續cmake方式安裝預設atlas,所以本人也用次

  sudo apt-get install libatlas-base-dev
  # openblas也很方便
  sudo apt-get install libopenblas-dev

Matlab的介面可以自己先安裝matlab ,此處省略,同時python可以安裝anconda來管理庫

安裝

Caffe

下載BVLC的caffe

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

編譯

1.Make方式

Make方式為官方的編譯方式,但是在後續caffe的使用時會稍顯麻煩,這裡要注意根據安裝的庫以及自己是否使用gpu、cudnn以及bals的選擇等作出修改

  cp Makefile.config.example Makefile.config
  # For CPU & GPU accelerated Caffe, no changes are needed.
  #For cuDNN acceleration using NVIDIA’s proprietary cuDNN software, uncomment the USE_CUDNN := 1 switch in #Makefile.config. cuDNN is sometimes but not always faster than Caffe’s GPU acceleration.
  #For CPU-only Caffe, uncomment CPU_ONLY := 1 in Makefile.config.
  # Adjust Makefile.config (for example, if using Anaconda Python, or if cuDNN is desired)
  make all -j8
  make test
  make runtest

2.Cmake方式

Cmake方式針對自己使用Caffe以及從軟體源安裝Caffe的使用者來說簡直不要更方便.

  mkdir build
  cd build
  cmake ..
  make all
  make install
  make runtest

由於自己使用Caffe不僅僅是停留在訓練,可能很多都要具體的測試實際的專案,因此相比於Make方式,Cmake的優勢就大大體現出來了。具體例子可以在我的github上看到https://github.com/zhanglaplace/MTCNN-Accelerate-Onet
編譯自己的專案,僅僅需要寫一個簡單的CMakeLists.txt檔案,並且檔案內的內容可以保證百分之九十的不變,這使得驗證演算法和專案變得相當方便.(強烈推薦)

  cmake_minimum_required(VERSION 2.9)
  project(MTCNN_Accelerate-Onet)  // 根據自己工程名字修改

  #set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
  set(CMAKE_CXX_STANDARD 11)

  find_package(OpenCV)

  find_package(Caffe REQUIRED)
  #message(FATAL_ERROR ${Caffe_INCLUDE_DIRS})
  include_directories(${Caffe_INCLUDE_DIRS})

  set(SOURCE_FILES main.cpp mtcnn.cpp mtcnn.h) // 根據自己實際原始碼修改
  add_executable(MTCNN_Accelerate-Onet ${SOURCE_FILES})

  target_link_libraries(MTCNN_Accelerate-Onet ${OpenCV_LIBS} )
  target_link_libraries(MTCNN_Accelerate-Onet ${Caffe_LIBRARIES})

相關推薦

Ubuntu快速配置Caffe

Caffe安裝 實際上在windows上安裝過多次caffe了,無論是BLVC版本的還是Microsoft版本的,ubuntu的按照也進行過,這段時間在自己筆記本上 又折騰了下caffe安裝,發現其實直接照著官方的是最方便快捷的。 具體可以參照 Installation instructions 具體步驟 根

Ubuntu安裝配置redis

命令 rep -c col 安裝配置 pan 令行 .html cli 安裝redis apt-get install redis-server 查看是否啟動 ps -aux|grep redis 客戶端連接   註:     安裝Redis服務器,會自動地一起安裝

ubuntuIDEA配置tomcat報錯Warning the selected directory is not a valid tomcat home

warn 技術 span 更改 文件目錄 tor 選擇 spa .com 產生這個問題的主要原因是文件夾權限問題。 可以修改文件夾權限或者更改tomcat文件目錄所有者。 這裏我直接變更tomcat文件夾所有者: sudo chown -R skh:skh tomcat-

Ubuntu 新手快速配置

軟件源 hive 裝包 但是 linux rom ins fire 文件 本渣渣剛剛開始學習linux,選擇Ubuntu作為入手系統,簡單總結了一下安裝常用軟件 安裝Vim linux上當然斯一件事就是安裝vim sudo apt-get install vim 安裝Chr

Nodejs 和 Electron ubuntu快速安裝

配置問題 選擇 https 前端知識 VM 失敗 url ctr 網上 查找時間管理軟件的時候發現了superProductivity這個程序,使用electron進行開發,於是看了一下介紹,手癢了,嘗試進行環境搭建,一下步慢慢補齊前端知識吧 nodejs安裝 nod

Windows 10安裝配置Caffe並支持GPU加速(改)

nvi 基本 一個 應該 添加 它的 右鍵 分享圖片 vid 基本環境 建議嚴格按照版本來 - Windows 10 - Visual Studio 2013 - Matlab R2016b - Anaconda - CUDA 8.0.44 - cuDN

ubuntuQt配置qwt和libmodbus

Qt安裝在ubuntu上出了很多問題,其中一個就是qmake不能用。 解決方法:進入/usr/lib/x86_64-linux-gnu/qt-default/qtchooser目錄,對目錄下的default.conf修改為 /home/mlw/Qt5.7.0/5.7/gcc_64/bin /

【GitHub】 初學Git,Github在Ubuntu配置

1.參考連結:https://blog.csdn.net/tina_ttl/article/details/51326684 http://www.runoob.com/w3cnote/git-guide.html 開始使用github 1. 配置git git config --g

快速配置caffe-ssd及如何配置環境變數路徑

caffe-ssd是目標檢測的一個經典模型,屬於單階段檢測模型,速度較快,本人實現過好幾次,也實現過ssd的改進版本fssd和DSSD,效果能提升幾個百分點。本篇部落格主要介紹caffe-ssd的配置流程以及如何設定環境變數路徑。 第一步:配置好caffe,包括NVIDIA

Ubuntu18.04安裝配置Caffe-SSD-GPU版本並MNIST模型測試和matlab caffe介面測試

Ubuntu18.04->sogou->顯示卡驅動->matlab2018a->cuda9.0->cudnn5.01->opencv3.4.1->caffe opencv 和caffe 比較難裝。其中opencv最為慢,主要make 一次太久。

Opencv 2.4.9在Ubuntu配置與安裝

一、依賴包安裝        1.  build-essential 軟體包,為編譯程式提供必需軟體包的列表資訊,這樣軟體包才知道標頭檔案、庫函式在哪裡。還會下載依賴的軟體包,安裝gcc/g++/gdb/make 等基本程式設計工具,最後才組成一個開發環境。 sudo a

Ubuntu18.04安裝配置Caffe-SSD-GPU版本

系統版本:Ubuntu18.04 系統環境:OpenCV3.4.1,Python2 1. 安裝依賴項 sudo apt-get install libprotobuf-dev liblevel

ubuntu如何配置靜態ip

前些天在xubuntu 18.04 desktop上設定了下靜態IP,發現與centos/linux中區別極大。 1、首先檢視網絡卡名 $ ip a / ifconfig 2、 cp一份yaml檔案,改個名字 $ sudo cp /etc/netplan/*.yaml /etc/

ubuntuapache2配置二級域名繫結專案資料夾

首先拿自己的域名生成一個二級域名然後上到自己的騰訊雲伺服器,找到Apache的配置檔案下面:進到available(準備就緒的那個資料夾)然後我們新建一個檔案仿照000-default.conf新建一個配置檔案之後進入000-default.conf下面將裡面的內容複製出來:

redis4.0.9 ubuntu 安裝配置

Redis下載地址:https://redis.io/download選擇穩定版下載:如果沒有安裝gcc,需要先安裝:sudo apt-get install gcc完了之後,解壓redis 壓縮包sudo tar -zxvf redis-4.0.9.tar.gz進入到 re

ubuntu安裝配置nodejs

nodejs是一個為javascript而生的強大“虛擬機器”,因此首選在虛擬機器下進行安裝執行,能更好的適應在原生的伺服器環境下進行開發。 常見的基本都是四種方法 1. 從原始碼安裝Node.JS 2. 用包管理器安裝Node.JS 3. 從

Ubuntu安裝配置NDK、獨立交叉編譯工具編譯C/C++程式

Android中IDE、ADT、SDK、JDK、NDK的解釋 1. IDE:Intelligent Development Environm的簡稱。即智慧開發環境。是一種開發工具。常用的IDE有adt-bundles和Android studio。兩個都需要配置jdk

Linux快速配置SSH免密碼登入

有機器A(192.168.221.128),B(192.168.221.129)。現想A和B想要通過SSH免密碼登入。 安裝openssh 首先我們需要先檢視本機是否已安裝openssh伺服器(openssh-server)或者客戶端(openssh-cli

windowsVS2015配置caffe

Windows下VS2015配置caffe,是一件其實挺簡單但是坑特別多的事情,稍有不慎就爬不出來,寫部落格來記錄最後在老師幫助下才配置成功的經驗(嚴格按照老師的教程一步也不能錯才行),當然我寫得也詳細一點,也讓大家少掉坑。 前提(開發環境):1、安裝完成VS2015

Ubuntu安裝配置防火墻

processor 電腦 class ssh 足夠 sed sso isa 條目 Ubuntu下安裝配置防火墻sudo apt-get install ufw 一般用戶,只需如下設置:sudo apt-get install ufw sudo ufw enable sud