Ubuntu18自带python版本为3.6.8,本文流程为
- 系统环境配置caffe(python3.6)
- miniconda3的3.6虚拟环境下运行caffe不报错
1.cuda+miniconda+cudnn+nvidia driver
2.系统环境配置caffe
2.1 安装caffe
根据caffe官网,版本高于17.04的ubuntu可以直接安装预编译的caffe,不用分别安装依赖了。
在命令行输入(不是conda里)
sudo apt install caffe-cuda #Gpu版本
sudo apt install caffe-cpu #Cpu版本
2.2 下载源码
cd到准备好的文件夹中,从github上下载caffe的源码
cd ***
git clone https://github.com/BVLC/caffe.git
2.3修改MakeFile.config
cd caffe
sudo cp Makefile.config.example Makefile.config
sudo vim Makefile.config #没有图形界面
sudo gedit Makefile.config #有图形界面
有图形界面直接修改,没有的话使用vim(按i之后修改,按esc退出修改模式,按:wq!保存退出)
修改位置如下
- 第4行
# cuDNN acceleration switch (uncomment to build with cuDNN).
USE_CUDNN := 1
- 第22行
# Uncomment if you're using OpenCV 3
OPENCV_VERSION := 3