主要参考文献:https://blog.csdn.net/CAU_Ayao/article/details/83536320
1、终端输入
sudo apt install caffe-cuda
这个语句能够在Ubuntu18.04上安装还caffe
但是安装的caffe不完善
2、安装依赖库
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev
sudo apt-get install 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
3、首先我们要从GitHub的远端下载caffe的源码
git clone https://github.com/BVLC/caffe.git
4、配置Makefile.config文件
cd caffe
sudo cp Makefile.config.example Makefile.config
sudo vim Makefile.config
进入Makefile.config后,点击 i 可以进行编辑,点击ESC退出,输入:wq 保存并退出
USE_CUDNN:= 1 (取消这行的注释)
OPENCV_VERSION :=3 (取消这行的注释)
BLAS:= atlas 注销, 换成 BLAS := open.
将 这两行删除掉 -gencode arch=compute_20,code=20 \
-gencode arch=compute_21,code=sm_21 \
删除后剩余的部分是这样子的:
CUDA_ARCH := -gencode arch=compute_30,code=sm_30 \
-gencode arch=compute_35,code=sm_35 \
-gencode arch=compute_50,code=sm_50 \
-gencode arch=compute_52,code=sm_52