tf2.x怎么禁用gpu

你可以使用 tf.config.set_visible_devices 来禁用 GPU。例如：

import tensorflow as tf

# 禁用所有 GPU
tf.config.set_visible_devices([], 'GPU')

# 禁用某个特定的 GPU
tf.config.set_visible_devices([], 'GPU:0')

这里的 'GPU' 表示所有 GPU，而 'GPU:0' 表示第一个 GPU(从 0 开始编号)。

请注意，这只是隐藏了 GPU，并不能真正地禁用 GPU。如果你希望完全禁用 GPU，你可以在环境变量