深度学习机器Caffe环境搭建 (centos 7)

系统 ： centos 7

• General dependencies

sudo yum install protobuf-devel leveldb-devel snappy-devel opencv-devel boost-devel hdf5-devel
#Remaining dependencies, recent OS

sudo yum install gflags-devel glog-devel lmdb-devel

• CUDA 安装

sudo rpm -i cuda-repo-rhel7-10-1-local-10.1.168-418.67-1.0-1.x86_64.rpm
export LD_LIBRARY_PATH=/usr/local/cuda
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH

• cuDNN 安装

Safari下载 https://developer.nvidia.com/cudnn

cp include/* /usr/local/cuda/include
cp lib64/* /usr/local/cuda/lib64

安装BLAS
选择安装ATLAS，安装命令如下：

sudo yum install atlas-devel 
cd  /usr/lib64/atlas 
sudo ln -sv libsatlas.so.3.10 libcblas.so
sudo ln -sv libsatlas.so.3.10 libatlas.so

caffe源码下载

#从github下载Caffe源码
git clone https://github.com/BVLC/caffe.git

#安装python 依赖
#安装前更新pip
sudo pip install --upgrade pip

#进入caffe/python目录，安装requirements中依赖库
for req in $(cat requirements.txt); do sudo pip install $req; done

• Caffe安装

make all -j2

• Pycaffe安装

make pycaffe

• 环境变量配置

#在~/.bashrc 的最后添加

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

export CUDA_HOME=/usr/local/cuda

• 安装成功
  

安装中遇到问题：
1.

原因在Makefile.config中找到

问题2:

/usr/bin/ld: 找不到 -lcudnn
collect2: 错误：ld 返回 1

解决 ：

cd /usr/local/cuda/lib64
#文件不存在libcudnn.so
sudo ln -sf libcudnn.so.7.4.1 libcudnn.so
sudo ln -s libcudnn.so.7.4.1 libcudnn.so.7
sudo ldconfig
#重新编译
make clean
make all -j2

问题3:

ERROR: Cannot uninstall ‘pyparsing’. It is a distutils installed project and thus we cannot accurately determine which files belong to it which would lead to only a partial uninstall.

解决

sudo pip install --ignore-installed pyparsing