没有GPU的机器上的TENSORFLOW,只能在CPU上运行它

Question

我正在学习Tensorflow(希望将它与我的聊天机器人集成),并希望运行此处的示例:

https://www.tensorflow.org/tutorials/recurrent

我下载了建议的存储库,安装了tensorflow(现在是cpu-only版本,但是也试过了--gpu)但是我坚持这个:

 python ptb_word_lm.py --data_path=simple-examples/data/ --model=small --rnn_mode=basic
2017-09-21 17:38:09.856638: W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.1 instructions, but these are available on your machine and could speed up CPU computations.
2017-09-21 17:38:09.856685: W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.2 instructions, but these are available on your machine and could speed up CPU computations.
Traceback (most recent call last):
  File "ptb_word_lm.py", line 532, in <module>
    tf.app.run()
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/platform/app.py", line 48, in run
    _sys.exit(main(_sys.argv[:1] + flags_passthrough))
  File "ptb_word_lm.py", line 459, in main
    % (len(gpus), FLAGS.num_gpus))
ValueError: Your machine has only 0 gpus which is less than the requested --num_gpus=1.

我的系统没有CUDA GPU(好吧,它有,但CUDA 1.3,而不是张量流需要的> = 3),如何让示例代码以cpu-only模式运行而不是仅仅退出？

Answer 1

您可以尝试将gpus的数量设置为您拥有的数字,即零.

python ptb_word_lm.py --data_path=simple-examples/data/ --model=small --rnn_mode=basic --num_gpus=0

我从错误消息中收集了这个:

ValueError: Your machine has only 0 gpus which is less than the requested --num_gpus=1.

由于您只有0gpus,因此需要明确设置该命令行arg,因为默认情况下是1在使用gpu包时.