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)