删除操作图张量流以在 CPU 上运行

Question

我已经训练了一个网络（使用 GPU），现在我想在 CPU 上运行它（用于推理）。为此，我使用以下代码加载元图，然后加载网络参数。

config = tf.ConfigProto(

        device_count = {'GPU': 0}

    )

sess = tf.Session(config=config)

meta_graph=".../graph-0207-190023.meta"

model=".../model.data-00000-of-00001"

new_saver = tf.train.import_meta_graph(meta_graph)

new_saver.restore(sess, model)

问题是，由于已经定义了用于训练的图，因此我使用了一些不在 CPU 上运行的特定操作。例如，“MaxBytesInUse” https://www.tensorflow.org/api_docs/python/tf/contrib/memory_stats/MaxBytesInUse其中记录了GPU的活动。

这就是为什么当我尝试运行此代码时，出现以下错误：

InvalidArgumentError: No OpKernel was registered to support Op 'MaxBytesInUse' with these attrs.  Registered devices: [CPU], Registered kernels:

  device='GPU'

     [[Node: PeakMemoryTracker/MaxBytesInUse = MaxBytesInUse[_device="/device:GPU:0"]()]]

是否有一种简单的方法可以删除特定的 GPU 相关操作并在 CPU 上运行图形？