100字范文 > caffe-gpu ubuntu 安装_ubuntu16.04 cuda10.0 配置caffe gpu环境

caffe-gpu ubuntu 安装_ubuntu16.04 cuda10.0 配置caffe gpu环境

时间：2020-11-08 20:39:02

caffe gpu环境的配置就是一部赤裸裸的血泪史，每次都能遇到很多问题~

环境依赖

ubuntu 16.04cuda 10.0python 3.5opencv 3.4.10sudo 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 sudo apt-get install git cmake build-essential

git clone /BVLC/caffe.gitcd caffecp Makefile.config.example Makefile.config

修改Makefile.config

１．# USE_CUDNN := 1 # USE_OPENCV := 0# OPENCV_VERSION := 3# WITH_PYTHON_LAYER := 1# 改为：USE_CUDNN := 1USE_OPENCV := 1OPENCV_VERSION := 3WITH_PYTHON_LAYER := 1２．INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/includeLIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib改为：INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serialLIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial３．#推荐使用python3.5的版本，一般Ubuntu系统自带的就行，python2.7太老，python3.6或者3.7容易出问题# 把python2.7代码加注释，python3.5前的注释去掉PYTHON_LIBRARIES := boost_python3 python3.5mPYTHON_INCLUDE := /usr/include/python3.5m \/usr/lib/python3.5/dist-packages/numpy/core/include４．# 如果cuda版本大于等于9.0，需要注释掉前两行，如下# -gencode arch=compute_20,code=sm_20 \#-gencodearch=compute_20,code=sm_21\CUDA_ARCH:=-gencodearch=compute_30,code=sm_30\ -gencodearch=compute_35,code=sm_35\ -gencodearch=compute_50,code=sm_50\ -gencodearch=compute_52,code=sm_52\ -gencodearch=compute_60,code=sm_60\ -gencodearch=compute_61,code=sm_61\ -gencodearch=compute_61,code=compute_61

修改Makefile

NVCCFLAGS += -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)改为：NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)LIBRARIES += glog gflags protobuf boost_system boost_filesystem m改为：LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial

开始编译

make all -j8

这里遇到一个报错:

cannot find -lboost_python3

解决方案如下(/BVLC/caffe/issues/4843)

cd/usr/lib/x86_64-linux-gnusudoln-slibboost_python-py35.solibboost_python3.so

报错解决之后make clean一下，重新编译

sudo make all-j8sudo make test -j8sudo make runtest -j8

sudo make all的结果

sudo make test的结果

sudo make runtest的结果

如果你的运行结果和上面的一致，那么恭喜你，让无数人蛋疼的caffe-gpu环境配置的深坑成功让你淌过去了，接下来测试一下。

mnist数据集测试

# 下载mnist数据集cd ~/caffe/data/mnist sudo sh ./get_mnist.sh# 在该目录下将有相应图片和标签文件mnist数据格式转换cd ~/caffe sudo sh ./examples/mnist/create_mnist.sh# 将在mnist文件夹(上个步骤的路径)生成LMDB格式数据集# 训练mnistcd ~/caffe sudo sh ./examples/mnist/train_lenet.sh

测试结果如下，到此caffe成功安装完毕

配置python接口

#配置环境变量sudo echo export PYTHONPATH="{caffe所在路径}/python" >> ~/.bashrc# 使之生效source ~/.bashrc

测试报错，缺少包

在这里卡了一下，系统目前默认python解释器是python3.5没问题，但pip安装的时候不知道为啥老是自动装到python3.6的环境中了

解决方案：

python -m pip install numpy scikit-image

如果默认python版本不是3.5，这里就需要指定具体的路径了，比如

/usr/bin/python3.5 -m pip install scikit-image

参考:

//06/install-the-pip-package-to-the-specified-python-environment/

最终测试