Advertisement

在Ubuntu 16.04上安装caffe Python接口

阅读量:

下载caffe:

git clonehttps://github.com/BVLC/caffe

安装opencv,具体步骤可以参考:

http://docs.opencv.org/2.4/doc/tutorials/introduction/linux_install/linux_install.html

Copy the Makefile.config.example to Makefile.config like this:

复制代码
    cp Makefile.config.example Makefile.config

编辑Makefile.config文件:

如果只使用CPU计算,修改:

去掉CPU_ONLY := 1前面的#

即修改为:

CPU_ONLY := 1

如果CUDA版本为 CUDA is 8.0,则需要将:

CUDA_DIR := /usr/local/cuda

更改为:

CUDA_DIR := /usr/local/cuda-8.0

安装python依赖库:

进入python目录:

cd caffe/python

运行安装命令:

复制代码
    for req in $(cat requirements.txt); do sudo pip install $req; done

编译caf

全部评论 (0)

还没有任何评论哟~