以 nvidia/cuda:9.0-cudnn7-devel-centos7 为基础镜像
基础环境 centos7 cuda9 cudnn7
安装anaconda3
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.3.0-Linux-x86_64.sh -O ~/anaconda.sh && \
/bin/bash ~/anaconda.sh -b -p /home/root/anaconda3 && \
rm ~/anaconda.sh && \
echo "export PATH=/home/root/anaconda3/bin:$PATH" >> ~/.bashrc
安装opencv
yum install opencv-devel
pkg-config --modversion opencv
如果输出了opencv2.4 的版本信息,说明安装成功
安装boost_1_58_0 cmake-3.10.2
安装caffe依赖
yum install protobuf-devel leveldb-devel snappy-devel opencv-devel boost-devel hdf5-devel
yum install gflags-devel glog-devel lmdb-devel
yum install openblas-devel
pip install numpy
pip install pandas
下载caffe 在/openpose/3rdparty
git clone https://github.com/BVLC/caffe.git
sudo cp Makefile.config.example Makefile.config
在文件中替换一下几个地方:
将
#USE_CUDNN := 1
修改成:
USE_CUDNN := 1
...
#如果此处是OpenCV2,则不用修改
将
#OPENCV_VERSION := 3
修改为:
OPENCV_VERSION := 3
...
将
#WITH_PYTHON_LAYER := 1
修改为
WITH_PYTHON_LAYER := 1
...
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
修改为:
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
...
CUDA_ARCH := -gencode arch=compute_20,code=sm_20 \
-gencode arch=compute_20,code=sm_21 \
-gencode arch=compute_30,code=sm_30 \
-gencod