tensorflow - 需要 tensorflow-gpu 时“没有名为 tensorflow 的模块”

Question

我构建了一个 python tensorflow 包并上传以在 ml 引擎上运行。

在我的 setup.py 中设置了“tensorflow-gpu==1.8.0”（无 tensorflow）。

ML 引擎运行在“import tensorflow as tf”时失败，说“没有名为 tensorflow 的模块”。

当我只需要“tensorflow==1.8.0”时，ML 引擎运行良好，但我相信使用 GPU 需要 tensorflow-gpu。

任何想法如何解决这个问题？

谢谢

标签： tensorflowgoogle-cloud-ml

Answer 1

当您有多个 python 版本时也会发生这种情况。在这种情况下，您必须为 tf 安装指定相关的 python 版本。例如，“python3 setup.py”而不是“python setup.py”。