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

Posted 2023-03-13

【中文标题】在 Tensorflow 的 DNNClassifier 估计器中记录设备信息

【英文标题】：Log device info in DNNClassifier estimator in Tensorflow

【发布时间】：2018-06-12 01:12:15

【问题描述】：

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

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

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