Question

我已经将anaconda更新为最新版本。它同时安装了tensorflow 2.2和tensorflow-gpu 2.2。但是，当我import tensorflow时，tensorflow-gpu是要使用的默认值。有没有办法在它们之间切换？

在此网站上询问的类似问题全部针对1.x版本。我已经尝试了解决方案，但似乎不适用于2.x版本。 Tensor flow toggle between CPU/GPU

Answer 1

这应该可以解决您的问题：

with tf.device('/gpu:0'):
    # YOUR def main() OR model.fit()

with tf.device('/cpu:0'):
    # YOUR def main() OR model.fit()

这应该适用于不带会话的TF2。

Answer 2

另一种方法是：

#use gpu 0
physical_devices = tf.config.list_physical_devices('GPU')
tf.config.set_visible_devices(physical_devices[0], 'GPU')

#use cpu
tf.config.set_visible_devices([], 'GPU')

对于tensorflow 2.x，如何在CPU和GPU版本之间切换？

2 个答案: