【发布时间】:2017-04-20 17:15:06
【问题描述】:
我使用 keras 和 tensorflow 后端构建顺序模型。 正如您在图片中看到的,模型使用所有内核,但没有一个是 100%。 我想知道,如何让keras+tf 100%利用每个核心。
我尝试从头开始构建 tensorflow 并重新安装,但没有帮助。
【问题讨论】:
-
你训练哪个网络?如果有很多全连接或卷积层,我认为 CPU 使用率会是 100%。
-
@pgplus1628 有 3 层 + 批量标准化的顺序模型。它的训练速度很慢,所以我认为应该是 100% 的使用率。
-
性能是否会受到数据加载的限制
-
@pgplus1628 不,我检查了 I/O 活动,它大约是 0。
标签: python machine-learning tensorflow neural-network keras