"No module named tensorflow" 当需要 tensorflow-gpu 时
"No module named tensorflow" when tensorflow-gpu is required
我在 ml 引擎上构建了一个 python tensorflow 包并上传到 运行。
"tensorflow-gpu==1.8.0"（无 tensorflow）在我的 setup.py.
中设置为必需
ML 引擎 运行 在 "import tensorflow as tf" 处失败，表示 "No module named tensorflow"。
当我只需要 "tensorflow==1.8.0" 时，ML 引擎 运行 工作正常，但我认为使用 GPU 需要 tensorflow-gpu。
有解决这个问题的想法吗？
谢谢
提交作业时需要设置--runtime-version=1.8。因此，您无需在 setup.py 中手动指定 TF。事实上，如果这是您唯一需要的包，您可以完全省略 setup.py。
更新 2018/06/29:
说明：不同版本的TensorFlow需要不同版本的NVIDIA驱动和软件栈。 --runtime-version 保证为特定版本的 TensorFlow 提供正确版本的驱动程序。从技术上讲，您可以在 setup.py 中设置 tensorflow-gpu 的版本，但该版本必须与您选择的 --runtime-version 中存在的 NVIDIA 堆栈兼容（默认为非常旧的 TF 1.0 ).
当您有多个 python 版本时也会发生这种情况。在这种情况下，您必须为 tf 安装指定相关的 python 版本。例如，"python3 setup.py" 而不是 "python setup.py"。
我在 ml 引擎上构建了一个 python tensorflow 包并上传到 运行。
"tensorflow-gpu==1.8.0"（无 tensorflow）在我的 setup.py.
中设置为必需ML 引擎 运行 在 "import tensorflow as tf" 处失败，表示 "No module named tensorflow"。
当我只需要 "tensorflow==1.8.0" 时，ML 引擎 运行 工作正常，但我认为使用 GPU 需要 tensorflow-gpu。
有解决这个问题的想法吗？
谢谢
提交作业时需要设置--runtime-version=1.8。因此，您无需在 setup.py 中手动指定 TF。事实上，如果这是您唯一需要的包，您可以完全省略 setup.py。
更新 2018/06/29:
说明：不同版本的TensorFlow需要不同版本的NVIDIA驱动和软件栈。 --runtime-version 保证为特定版本的 TensorFlow 提供正确版本的驱动程序。从技术上讲，您可以在 setup.py 中设置 tensorflow-gpu 的版本，但该版本必须与您选择的 --runtime-version 中存在的 NVIDIA 堆栈兼容（默认为非常旧的 TF 1.0 ).
当您有多个 python 版本时也会发生这种情况。在这种情况下，您必须为 tf 安装指定相关的 python 版本。例如，"python3 setup.py" 而不是 "python setup.py"。