发新帖

tensorflow用于线上预测的参数设置

[复制链接]
226 2

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
想用java版tensorflow做线上预测,发现gpu耗时比cpu还多,想请教一下参数应该如何设置。如session的个数,session的参数设置,线上每台机器是一张显卡
我知道答案 回答被采纳将会获得10 金币 + 10 金币 已有2人回答
本楼点评(0) 收起

精彩评论2

Antonio  TF荚荚  发表于 2018-12-18 15:10:41 | 显示全部楼层
理论上讲GPU应该比CPU快才对,你可以把中间步骤耗时打印出来看看,到底是哪个部分耗费了时间,或许并不是run的过程中耗时多了呢
本楼点评(0) 收起
落月星辰  TF荚荚  发表于 2018-12-22 14:09:26 来自手机  | 显示全部楼层
一开始是由CPU处理图片的加载,然后进入GPU内存进行运算,然后运算下来后还是CPU执行后续操作,或许是你的代码里面CPU的操作过多了,或者说是加载数据的方法有些问题,你可以看下纯CPU时间做一个对比
本楼点评(0) 收起
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

快速回复 返回顶部 返回列表