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

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值