2016-05-14 97 views
0

當我嘗試與啓-V3我自己的形象基地爲後續訓練:我可以用cpu訓練初始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 

我的問題是「是否有可能,我可以列車啓動-v3模型只與CPU?「 PS:我知道如果只使用比gpu花費很多時間的cpu,我會減少max_steps和圖像數據。

非常感謝!

回答

0

我自己解決了這個愚蠢的問題。 我已經買了GTX 960,謝謝大家。 我一步一步如下: http://blog.twman.org/2016/06/tensorflow.html

+1

很高興看到你解決了這個問題。請注意,純外部鏈接答案被視爲低質量答案,這是這些帖子被刪除的主要原因。爲了避免這種情況,請使用解決方案的重要部分編輯您的文章 - 來自評論(http://stackoverflow.com/review/late-answers/12952803) – Raju

相關問題