我可以只用 cpu 训练 inception-v3 模型吗？

Question

当我尝试在 Inception-v3 上使用自己的图像进行训练时，如下所示：

https://github.com/tensorflow/models/blob/master/inception/inception/inception_train.py

我收到这个错误：

File "/tensorflow/tensorflow/models/inception/bazel-bin/inception/flowers_train.runfiles/inception/inception_train.py", line 207, in train
    assert FLAGS.batch_size % FLAGS.num_gpus == 0, (
ZeroDivisionError: division by zero

我的问题是"Is it possible that I can train inception-v3 model only with cpu ?"

PS：我知道如果只使用 cpu 会比 gpu 花费更多时间，因此我减少了 max_steps 和图像数据。

非常感谢！

Answer 1

我已经自己解决了这个愚蠢的问题。买了GTX 960，谢谢大家。我的reslove一步一步如下： http://blog.twman.org/2016/06/tensorflow.html

我可以只用 cpu 训练 inception-v3 模型吗？

Is it possible that I can train inception-v3 model only with cpu?

tensorflow