在 Tensorflow 的 DNNClassifier 估计器中记录设备信息答案

【问题标题】：Log device info in DNNClassifier estimator in Tensorflow在 Tensorflow 的 DNNClassifier 估计器中记录设备信息
【发布时间】：2018-06-12 01:12:15
【问题描述】：

我正在使用DNNClassifier Estimator 来训练二元分类器。我想记录设备信息以验证我的模型是在 GPU 还是 CPU 上运行。

由于使用 Estimator 我们不处理 session，我如何记录设备信息？

主要问题：我的 3 带有隐藏单元 [100, 75, 50] 的分层神经网络在 CPU 上的运行速度比在 GPU 上快。我尝试将批量大小增加到 256，但仍然相同。因此，我想确认它是否真的在使用 GPU。

【问题讨论】：

标签： tensorflow neural-network deep-learning classification gpu

【解决方案1】：

使用tf.estimator.Estimator.__init__ 的config 参数：

classifier = \
  DNNClassifier(feature_columns=feature_columns,
                hidden_units=[100, 75, 50],
                config=tf.estimator.RunConfig(session_config=tf.ConfigProto(log_device_placement=True)))

【讨论】：