在（！）TensorFlow使用一次/ Reset TensorFlow之后切换CPU / GPU

Question

我正在尝试比较GPU与CPU上的TensorFlow图的运行时间。

我知道我可以通过更改环境变量CUDA_VISIBLE_DEVICES来启用/禁用GPU使用率。但是，如果我之前已经处理过任何TensorFlow计算，则此更改不会影响任何内容。

所以我可能需要重置TensorFlow 。不过，session.close()和tf.reset_default_graph()都不会对此有所帮助。

最后，我知道我可以用

之类的东西包装我的代码部分

 with tf.Session() as sess:
      with tf.device("/cpu:0"):

，但这只能在本地帮助，因此需要我更改很多代码部分。相反，我正在寻找一个全局解决方案，它不需要我开始一个全新的Python会话。

非常感谢任何帮助！