- 博客(1)
- 收藏
- 关注
RSS订阅原创 caffe 去掉GPU依赖
最近因为要将caffe移植到没有GPU的电脑上，需要去掉GPU依赖，首先下载caffe windows 版，跟踪函数发现在头文件device_alternate.hpp中需要定义一个cpu_only的宏，定义完成后layer层会自动去掉调用GPU的函数，并且每个layer的实现函数会将forward和backward的实现定义为no gpu，然后在分类的时候setmode为caffe::cpu,就
2016-11-03 16:07:16
693
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人