Skip to content

lichee/caffe-auto-install

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

##caffe自动安装 ###1. 安装普通依赖项,编译cpu-only caffe:

  • 使用sudo auto.sh来安装caffe需要的c++库以及python包.
  • auto.sh会在home目录下载caffe源代码并编译cpu-only版caffe

至此,caffe已经安装完成, 只是没有GPU支持, 跑大型网络很费时.

如果你的计算机安装有支持cuda的显卡, 请继续按一下步骤安装cuda ###2. 安装cuda, 编译caffe_with_gpu

  • 根据你的系统版本 下载 cuda离线安装包,在"Architecture"这个选项中, Intel x_64或者AMD x_64 CPU请选择x_86_64, IBM P8工作站请选择ppc64le, 最后install type选择deb(local)本地安装文件.
  • 使用sudo cuda.sh xxx.deb安装cuda, xxx.deb为上一步下载的安装包文件名.
  • 进入caffe目录将Makefile.config文件中的CPU_ONLY := 1注释掉.
  • 使用make命令重新编译caffe.

###其他说明

  • auto.sh默认安装了caffe所需要的所有依赖, 包括画网络结构图等其他非必须工具所依赖的python包, 所以安装会比较耗时.
  • 如果需要简便安装可使用lite.sh, 仅安装caffe编译时必要的c++库, 不安装python包, 你可以简便编译安装caffe, 视情况需要安装python包.
  • python包之间的依赖关系很复杂, 在ubuntu下强烈不建议使用apt-get install python-xx来安装管理python package, 推荐使用pip,Mac环境也适用.

如果使用中碰到问题, 欢迎邮件到 zeakey@outlook.com

About

auto-install caffe on ubuntu

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.9%
  • Shell 0.1%