[环境配置]ubuntu18.04+python3.6+caffe-gpu配置记录

Ubuntu18自带python版本为3.6.8,本文流程为

  • 系统环境配置caffe(python3.6)
  • miniconda3的3.6虚拟环境下运行caffe不报错

1.cuda+miniconda+cudnn+nvidia driver

首先,安装cuda等(之前写过,点这里)

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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值