我正在尝试比较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会话。
非常感谢任何帮助！