下载软件:
1. 安装caffe
$ git clone https://github.com/BVLC/caffe.git
2. 安装cuda(默认已安装)
配置参数:
$ cd caffe
$ cp Makefile.config.example Makefile.config
$ vim Makefile.config
对以下进行设置:
USE_CUDNN:= 1
CUDA_DIR:= /usr/lcoal/cuda
MATLAB_DIR := /usr/local/R2018a (如果需要matlab接口的话)
# PYTHON_INCLUDE := ....python2.7.. 把python2.7注释掉
PYTHON_LIBRARIES := boost_python3 python3.5m 把python3的配置去掉注释
PYTHON_INCLUDE := ..... 同上一行去掉注释
PYTHON_LIB := /usr/lib
WITH_PYTHON_LAYER := 1
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial