Ubuntu16.04 + cuda9.0 + cudnn7 于服务器上配置,参考前请对照自身需要。
安装依赖包
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 libopenblas-dev liblapack-dev libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
创建Makefile.config文件
cp Makefile.config.example Makefile.config
修改Makefile.config
1.第一步
a.如果要使用cudnn, 则将
#USE_CUDNN = 1
改为
USE_CUDNN = 1
b. 如果使用的opencv是版本3的, 则将
OPENCV_VERSION = 3
改为
OPENCV_VERSION = 3
c. 如果要使用python来修改layer,则将
WITH_PYTHON_LAYER = 1
改为
WITH_PYTHON_LAYER = 1
d.因为是cuda9,所以把
CUDA_ARCH = -gencode