tensorflow 怎么设置成cpu运行

2024-11-18 04:29:31
推荐回答(2个)
回答(1):

  基本使用
  使用 TensorFlow, 你必须明白 TensorFlow:
  使用图 (graph) 来表示计算任务。
  在被称之为 会话 (Session) 的上下文 (context) 中执行图。
  使用 tensor 表示数据。
  通过 变量 (Variable) 维护状态。
  使用 feed 和 fetch 可以为任意的操作(arbitrary oper搜索ation) 赋值或者从其中获取数据。

回答(2):

config = tf.ConfigProto(
device_count = {'CPU': 1}
)
sess = tf.Session(config=config)