如何在没有 GPU 的情况下运行 tensorflow？

Question

我的系统有一个 GPU。

当我在其上运行 Tensorflow 时，TF 会自动检测 GPU 并开始在 GPU 上运行线程。

我该如何更改？

即如何在没有 GPU 的情况下运行 Tensorflow？

Answer 1

它应该工作。它主要禁用 CUDA 设备。因此，代码会寻找其他来源(CPU) 来运行代码。

import os
import tensorflow as tf
#os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" #If it doesn't work, uncomment this line; it should help.
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
#Your Code Here