caffe 去掉GPU依赖

来源:互联网 发布:原油手机行情分析软件 编辑:程序博客网 时间:2024/05/29 10:50
最近因为要将caffe移植到没有GPU的电脑上,需要去掉GPU依赖,首先下载caffe windows 版,跟踪函数发现在头文件device_alternate.hpp中需要定义一个cpu_only的宏,定义完成后layer层会自动去掉调用GPU的函数,并且每个layer的实现函数会将forward和backward的实现定义为no gpu,预处理器里将use_cudnn去掉,然后在分类的时候setmode为caffe::cpu,就完成了
0 0