在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)
还没有任何评论哟~
