TensorFlow GPU比CPU更慢

发布于 2022-09-06 09:57:32 字数 245 浏览 16 评论 0

CPU: i7-4710HQ
显卡:GTX850m
内存:8G
操作系统:Win10
装了 CUDA8.0,cuDNN6.0,tensorflow-gpu-1.4.0

配置了必要的PATH环境变量之外没做其它配置了。

用同样的TensorFlow代码跑同一个BPNN,用CPU只花了32秒完成,用GPU却执行了140+秒。
为什么我的TensorFlow加了GPU优化之后反而更慢了?
是不是我参数没配置好?

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。

评论(1

感性不性感 2022-09-13 09:57:32

第二天上网Google了一番,有的说是可能网络太小,效率瓶颈卡在了CPU与GPU数据传输上。

对此我做了一波实验,原来的BPNN只有双隐层,节点数分别是15,5,我把改成了4隐层,每层节点数都是2000之后,对单次训练进行了时间统计。

实验出来这时GPU单次训练只花了30~50 ms左右,而CPU的平均耗时是500ms,由此可以断定一个结论:
网络结构比较小的时候,效率瓶颈在CPU与GPU数据传输,这个时候只用cpu会更快。
网络结构比较庞大的时候,gpu的提速就比较明显了。

~没有更多了~
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文