建站
咨詢
售后
建站咨詢
新聞中心
這里有您想知道的互聯(lián)網(wǎng)營銷解決方案
Ubuntu18.04安裝caffe具體步驟
caffe的全稱是Convolutional Architecture for Fast Feature Embedding（譯為：快速特征嵌入的卷積體系結(jié)構(gòu)），核心語言是C++。caffe的基本工作流程是設(shè)計建立在神經(jīng)網(wǎng)絡(luò)的一個簡單假設(shè)，所有的計算都是層的形式表示的，網(wǎng)絡(luò)層所做的事情就是輸入數(shù)據(jù)，然后輸出計算結(jié)果。
創(chuàng)新互聯(lián)是專業(yè)的嵐縣網(wǎng)站建設(shè)公司，嵐縣接單;提供成都做網(wǎng)站、成都網(wǎng)站制作,網(wǎng)頁設(shè)計,網(wǎng)站設(shè)計,建網(wǎng)站,PHP網(wǎng)站建設(shè)等專業(yè)做網(wǎng)站服務(wù);采用PHP框架,可快速的進(jìn)行嵐縣網(wǎng)站開發(fā)網(wǎng)頁制作和功能擴(kuò)展;專業(yè)做搜索引擎喜愛的網(wǎng)站,專業(yè)的做網(wǎng)站團(tuán)隊(duì),希望更多企業(yè)前來合作!
安裝環(huán)境
-
OS: Ubuntu 18.04 64 bit
-
顯卡: NVidia GTX 1080
-
CUDA: 10.0
-
cuDNN: 7.4
-
Python: 3.6
任務(wù)：從源碼編譯安裝caffe
安裝OpenCV
參考Ubuntu 18.04源碼編譯安裝OpenCV 4.0步驟，安裝OpenCV。
安裝依賴
$ apt install -y libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
$ apt install -y --no-install-recommends libboost-all-dev
$ apt install -y libatlas-base-dev
$ apt install -y libgflags-dev libgoogle-glog-dev liblmdb-dev
下載源碼
$ git clone https://github.com/BVLC/caffe.git
$ cd caffe
$ cp Makefile.config.example Makefile.config
更新代碼
因?yàn)槭褂?code>OpenCV 4.0的版本，所以有些常量需要更新：
CV_LOAD_IMAGE_GRAYSCALE -> cv::ImreadModes::IMREAD_GRAYSCALE
CV_LOAD_IMAGE_COLOR -> cv::IMREAD_COLOR
替換命令如下：
sed -i 's/CV_LOAD_IMAGE_COLOR/cv::IMREAD_COLOR/g' src/caffe/layers/window_data_layer.cpp
sed -i 's/CV_LOAD_IMAGE_COLOR/cv::IMREAD_COLOR/g' src/caffe/util/io.cpp
sed -i 's/CV_LOAD_IMAGE_GRAYSCALE/cv::ImreadModes::IMREAD_GRAYSCALE/g' src/caffe/util/io.cpp
修改配置
為了使用OpenCV 4.0 + CUDA 10 + cuDNN + python 3.6，修改Makefile.config如下：
...
USE_CUDNN := 1
...
OPENCV_VERSION := 3
...
CUDA_DIR := /usr/local/cuda-10.0
...
CUDA_ARCH := \
-gencode arch=compute_30,code=sm_30 \
-gencode arch=compute_35,code=sm_35 \
-gencode arch=compute_50,code=sm_50 \
-gencode arch=compute_52,code=sm_52 \
-gencode arch=compute_60,code=sm_60 \
-gencode arch=compute_61,code=sm_61 \
-gencode arch=compute_61,code=compute_61
...
# Uncomment to use Python 3 (default is Python 2)
PYTHON_LIBRARIES := boost_python3 python3.6m
PYTHON_INCLUDE := /usr/include/python3.6m \
/usr/local/lib/python3.6/dist-packages/numpy/core/include
...
WITH_PYTHON_LAYER := 1
...
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial /usr/local/cuda/include /usr/local/include/opencv4
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial
編譯caffe
$ make -j8
$ make pycaffe
一切順利的話，生成的工具在build/tools目錄下。
錯誤處理
如果報錯如下：
/usr/local/lib/libopencv_imgcodecs.so: undefined reference to `TIFFReadDirectory@LIBTIFF_4.0' 解決方案：
$ conda uninstall libtiff
文章題目：Ubuntu18.04安裝caffe具體步驟
文章路徑：http://www.5511xx.com/article/ccohipg.html
