Ubuntu 16.04 Anaconda3+opencv3.4.6+Caffe+多GPU+CUDA10.1
默认已经安装NVIDIA显卡驱动以及CUDA、cudnn
博主的安装环境是是Ubuntu16.04 、NVIDIA 2080TI最新显卡、CUDA10.1、CUDNN10.1
如果没有安装请自行解决相关安装
Anaconda安装
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-4.2.0-Linux-x86_64.sh #如果没有,则需要安装wget
bash Anaconda3-4.2.0-Linux-x86_64.sh #安装anaconda,一路yes
其他问题请查阅相关资料,建议添加中科大镜像,速度要快点、
Caffe下载
sudo apt install git
git clone https://github.com/BVLC/caffe.git #下载linux版的caffe
这里下载速度真的慢,可以直接去网站上下载,然后进行解压
相关依赖库安装
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install -y build-essential cmake git pkg-config
sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install -y libatlas-base-dev
sudo apt-get install -y --no-install-recommends libboost-all-dev
sudo apt-get install -y libgflags-dev libgoogle-glog-dev liblmdb-dev
这下面的操作能执行就执行,其实也只是为了规避一些报错,虽然下面还会报很多很多错误。。。。。。。。。
conda install libgcc
conda install protobuf
conda install -c menpo opencv3 #安装opencv3
opencv安装
这里根据CUDA版本问题,楼主的版本是CUDA10.1。
opencv3.2一直无法安装,所以建议安装opencv-3.4.6.在这里折腾了很久很久,亲测,安装3.4.6版本没有问题
- 进入官网 : http://opencv.org/releases.html , 选择 3.4.6 版本的 source , 下载 opencv-3.4.6.zip
- 解压到你要安装的位置,命令行进入已解压的文件夹 opencv-3.4.6目录下,执行:
mkdir build #创建编译的文件目录
cd build
cmake -D CMAKE_BUILD_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local ..
# cmake -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local -D FORCE_VTK=ON -D WITH_TBB=ON -D WITH_V4L=ON -D WITH_QT=ON -D WITH_OPENGL=ON -D WITH_CUBLAS=ON -D CUDA_NVCC_FLAGS="-D_FORCE_INLINES --expt-relaxed-constexpr" -D WITH_GDAL=ON -D WITH_XINE=ON -D BUILD_EXAMPLES=ON ..
make -j8 # CPU核数多就多设置一点
- 然后编译,编译成功后安装:
sudo make install #安装
- 安装完成后通过查看 opencv 版本验证是否安装成功:
pkg-config --modversion opencv
caffe文件修改
在此之前,我们得加入以下环境变量
cd /home/user
gedit ~/.bashrc
在文件末尾添加以下两行(注意 usr为博主自己用户名,笔者这里为amax,请根据情况自行修改)
export LD_LIBRARY_PATH=/home/amax/anaconda3/lib:$LD_LIBRARY_PATH
export PYTHONPATH=/home/amax/caffe/python:$PYTHONPATH
然后
source ~/.bashrc
然后进入caffe目录下
cd /home/user/caffe #进入caffe目录
cp Makefile.config.example Makefile.config
官方推荐用kate修改make文件,不过 gedit也行
sudo apt-get install kate
博主vi也用不惯,就用low b gedit了
接着
Makefile.config文件修改
sudo gedit Makefile.config
1.查找
USE_CUDNN := 1
USE_OPENCV := 1
OPENCV_VERSION := 3
WITH_PYTHON_LAYER := 1 (这一句大概在94行 自行查找 ctrl+F)
全部取消注释
39-45行修改,删除掉compute_20 21