在tensorflow中找不到device_util

Question

我正在尝试使用以下方式导入device_util：

from tensorflow.python.distribute import device_util

但是找不到device_util。版本是tensorflow-gpu 1.12.0。

Answer 1

它最近被移动了。

在training下试用1.12.0版：

from tensorflow.python.training import device_util

源代码：https://github.com/tensorflow/tensorflow/blob/v1.12.0/tensorflow/python/training/device_util.py