我在其他地方看到了一些建议，但它们很旧，不适用于较新的 TF 版本。对我有用的是：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

当该变量已定义且等于 -1 时，即使 CUDA GPU 可用，TF 也会使用 CPU。

对于 TF2：

try:
    # Disable all GPUS
    tf.config.set_visible_devices([], 'GPU')
    visible_devices = tf.config.get_visible_devices()
    for device in visible_devices:
        assert device.device_type != 'GPU'
except:
    # Invalid device or cannot modify virtual devices once initialized.
    pass

我发现在脚本之外设置变量最容易并且总是有效的。

export CUDA_VISIBLE_DEVICES=''

在运行 python 脚本之前在命令行上运行它。