GCP执行gcloud ai-platform local train命令提示无tensorflow模块问题咨询

核心原因 #

gcloud ai-platform local train 命令默认调用Google Cloud SDK自带的独立Python解释器，不会使用你当前终端激活的虚拟环境、或你自行安装的Python 3.7环境。你之前安装的tensorflow、numpy均在自己常用的Python环境下，SDK自带的Python环境中没有对应依赖，因此会触发模块不存在报错。该问题和GCP库的特殊安装路径无关。

问题触发的常见原因 #

• 本地使用venv、conda等工具管理Python虚拟环境，执行gcloud命令时，虚拟环境的Python解释器优先级未覆盖SDK自带的Python路径
• Mac设备下Google Cloud SDK默认安装路径为~/google-cloud-sdk，自带的Python解释器位于~/google-cloud-sdk/platform/bundledpythonunix/bin/python3，和用户自行安装的Python 3.7是完全隔离的运行环境
• 执行local train命令时未显式指定Python解释器路径，gcloud默认走自带的Python环境加载依赖

验证方案 #

执行以下命令即可确认gcloud当前调用的Python解释器路径：

gcloud info | grep Python

如果输出的路径不是你安装了tensorflow、numpy的Python 3.7路径，即可确认是环境不匹配导致的报错。

修复方案 #

• 方案1：单次执行命令时显式指定Python路径
  执行命令时添加--python-path参数指向你安装了依赖的Python 3.7路径，可先通过which python3.7查询对应路径，示例命令如下：

  gcloud ai-platform local train --python-version 3.7 --python-path /替换为你的Python3.7路径 --module-name 你的训练模块名 --package-path 你的代码包路径
  

• 方案2：设置gcloud默认Python解释器
  执行以下命令配置全局默认的local train Python路径，后续执行命令无需重复添加参数：

  gcloud config set ai_platform/local_python /替换为你的Python3.7路径
  

• 方案3：给SDK自带的Python环境安装依赖（不推荐）
  直接给SDK自带的Python解释器安装需要的依赖，后续更新SDK时依赖可能会被重置，命令参考：

  ~/google-cloud-sdk/platform/bundledpythonunix/bin/pip install tensorflow numpy
  

内容的提问来源于stack exchange，提问作者user3628719