我构建了一个 python tensorflow 包并上传到 ml 引擎上运行。
“tensorflow-gpu==1.8.0”（无 tensorflow）在我的 setup.py 中设置为必需。
ML 引擎运行在“import tensorflow as tf”时失败，说“没有名为 tensorflow 的模块”。
当我只需要“tensorflow==1.8.0”时，ML 引擎运行正常，但我相信 tensorflow-gpu 需要使用 GPU。
任何想法如何解决这个问题？
谢谢
--runtime-version=1.8提交作业时需要设置。因此，您无需在 setup.py 中手动指定 TF。事实上，如果这是你唯一需要的包，你可以完全省略 setup.py。
2018/06/29 更新：
说明：不同版本的 TensorFlow 需要不同版本的 NVIDIA 驱动程序和软件堆栈。在--runtime-version保证具有的驱动程序TensorFlow的特定版本的正确版本。从技术上讲，您可以tensorflow-gpu在 setup.py 中设置 的版本，但该版本必须与--runtime-version您选择的 NVIDIA 堆栈兼容（默认为非常旧的 TF 1.0）。
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句