在CPU上运行Tensorflow基准测试?

时间:2018-03-28 16:31:55

标签: python tensorflow

如何在CPU上运行Tensorflow CNN benchmarks?想与GPU比较。谢谢!

编辑:对于有此问题的其他人,我还必须添加--data_format=NHWC

1 个答案:

答案 0 :(得分:1)

使用命令行选项--device cpu