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，而不是张量流需要的＆gt; = 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.

由于您只有0 gpus，因此在使用gpu包时，需要明确设置该命令行arg，因为默认值为1。

没有GPU的机器上的TENSORFLOW，无法仅在CPU上运行

1 个答案: