我是否有可能只用cpu训练inception-v3模型?

时间:2016-05-14 19:15:02

标签: tensorflow

当我尝试使用我自己的图像基于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

我的问题是“我是否有可能只使用cpu训练初始-v3模型?”

PS:我知道如果只使用比gpu花费更多时间的cpu,我会减少max_steps和图像数据。

非常感谢!

1 个答案:

答案 0 :(得分:0)

我自己解决了这个愚蠢的问题。 我买了一台GTX 960,谢谢大家。 我一步一步地重新开始如下: http://blog.twman.org/2016/06/tensorflow.html