Python 代码在 Azure ML 实验中失败，但在 conda 环境中运行

Question

我已在 Azure ML 计算实例中创建了自定义 conda 环境，并已验证 python 代码是否在该环境中运行。但是，当我在 Azure ML 实验中提交 .py 文件时，即使为实验设置了相同的 conda 环境，运行也会失败。

这就是我提交实验的方式:

ws = Workspace.from_config()
compute_name = os.environ.get("AML_COMPUTE_CLUSTER_NAME", "mycompute_cluster")
compute_target = ws.compute_targets[compute_name]
env = Environment.from_existing_conda_environment('expEnv', "myEnv")
experiment = Experiment(workspace=ws, name='exp')
config = ScriptRunConfig(source_directory='./',
                             script='exp1.py',
                             compute_target=compute_target)

config.run_config.environment = env
run = experiment.submit(config)
aml_url = run.get_portal_url()
print(aml_url)

我还尝试从 conda YAML 文件创建 Azure ML 环境并在提交实验时使用它，但我仍然遇到相同的错误。

错误:

UserError: module 'tensorflow.python.training.experimental.mixed_precision' has no attribute '_register_wrapper_optimizer_cls'

Answer 1

所需的模块似乎丢失或重命名。您需要确保在代码中使用具有 required 属性的相同模块版本。

此外，您可能需要按照 @HarshithaVeeramalla-MT 在评论部分中的建议使用 pip3 install tf-nightly 命令安装模块 tf-nightly。