如何禁用 TensorFlow GPU？

Question

我首先在我的 GPU 上使用 python 创建了 TensorFlow 代码：

import tensorflow-gpu as tf

我将其用于培训目的，一切都很顺利。但现在，我想在没有 GPU 的设备上部署我的 python 脚本。所以，我用pip卸载了tensorflow-gpu并导入正常的TensorFlow：

import tensorflow as tf

但是当我运行脚本时，它仍在使用 GPU：

我尝试过这段代码：

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

但仍然无法工作（即使 GPU 似乎已禁用，如屏幕截图中的白色所示）。

我只想返回默认的 TensorFlow 安装，不带 GPU 功能。 我尝试卸载并安装tensorflow，删除虚拟环境并创建一个新环境，但没有任何效果。

感谢您的帮助！

Answer 1

conda，tensorflow 2.16.1（不是tensorflow[和-cuda]）

仅此

import tensorflow as tf
tf.config.set_visible_devices([], 'GPU')

没有这个

os.environ["CUDA_VISIBLE_DEVICES"] = ""