我如何 运行 没有 GPU 的 tensorflow?
How can I run tensorflow without GPU?
我的系统有一个 GPU。
当我 运行 Tensorflow 在上面时,TF 会自动检测 GPU 并在 GPU 上启动 运行ning 线程。
我该如何更改?
即我如何 运行 没有 GPU 的 Tensorflow?
应该可以。它主要禁用 CUDA 设备。因此,代码寻找其他来源 (CPU)
到 运行 代码。
import os
import tensorflow as tf
#os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" #If the line below doesn't work, uncomment this line (make sure to comment the line below); it should help.
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
#Your Code Here
我的系统有一个 GPU。
当我 运行 Tensorflow 在上面时,TF 会自动检测 GPU 并在 GPU 上启动 运行ning 线程。
我该如何更改?
即我如何 运行 没有 GPU 的 Tensorflow?
应该可以。它主要禁用 CUDA 设备。因此,代码寻找其他来源 (CPU)
到 运行 代码。
import os
import tensorflow as tf
#os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" #If the line below doesn't work, uncomment this line (make sure to comment the line below); it should help.
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
#Your Code Here