本文主要是介绍在Ubuntu17.10上使用CPU运行caffe的SSD,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
因为实验室的GPU还没采购好,只能在CPU上用别人训练好的模型跑几个简单的demo。先在台式机上装个CPU版的Caffe。网上很多教程讲的不是很详细,我就写了个完整版。
准备工作
安装caffe的依赖
非常不建议使用源码编译安装Caffe的依赖,因为这过程中可能会出现新的依赖问题。caffe官网推荐采用apt获取编译所需的依赖。
参考http://caffe.berkeleyvision.org/install_apt.html
sudo apt build-dep caffe-cpu
对于Ubuntu17.04及以上版本,官方提供了caffe的源码编译环境,但是实际上在后面编译的时候可能会提示并未找到某些依赖,如果出现找不到依赖的情况,根据报错补装。- 编译caffe所需要的所有依赖
- 基本依赖
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
- CUDA高性能并行计算库:我们安装的是CPU版本,不存在CUDA
- BLAS线性代数库,有两种开源线性代数库可供选择:
- atlas
sudo apt-get install libatlas-base-
- atlas
- 基本依赖
这篇关于在Ubuntu17.10上使用CPU运行caffe的SSD的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!