如何在azureml.core.runconfig.DockerConfiguration类或azureml.core.Environment类中使用azureml.core.ScriptRunConfig类

Question

我使用Microsoft Azure Machine Learning (Azure-ml)运行我的(python)实验。

对于指定VM和python环境，我使用：

from azureml.core import Environment
from azureml.core import ScriptRunConfig

# Other imports and code...

# Specify VM and Python environment:
vm_env = Environment.from_conda_specification(name='my-test-env', file_path=PATH_TO_YAML_FILE)
vm_env.docker.enabled = True
vm_env.docker.base_image = 'mcr.microsoft.com/azureml/openmpi3.1.2-cuda10.2-cudnn7-ubuntu18.04'

# Finally, use the environment in the ScriptRunConfig:
src = ScriptRunConfig(source_directory=DEPLOY_CONTAINER_FOLDER_PATH,
                      script=SCRIPT_FILE_TO_EXECUTE,
                      arguments=EXECUTE_ARGUMENTS,
                      compute_target=compute_target,
                      environment=vm_env)

我收到关于行vm_env.docker.enabled = True的以下警告

'enabled' is deprecated. Please use the azureml.core.runconfig.DockerConfiguration object with the 'use_docker' param instead.

关于DockerSection Class和DockerConfiguration Class的文档对于应用DockerConfiguration Class并不十分清楚。

我不知道如何使用azureml.core.runconfig.DockerConfiguration 对象。有人能给我举个例子吗？谢谢你！

Answer 1

ScriptRunConfig类现在接受一个docker_runtime_config参数，这是传递DockerConfiguration对象的地方。

所以，代码应该是这样的：

from azureml.core import Environment
from azureml.core import ScriptRunConfig
from azureml.core.runconfig import DockerConfiguration

# Other imports and code...

# Specify VM and Python environment:
vm_env = Environment.from_conda_specification(name='my-test-env', file_path=PATH_TO_YAML_FILE)
vm_env.docker.base_image = 'mcr.microsoft.com/azureml/openmpi3.1.2-cuda10.2-cudnn7-ubuntu18.04'

docker_config = DockerConfiguration(use_docker=True)

# Finally, use the environment in the ScriptRunConfig:
src = ScriptRunConfig(source_directory=DEPLOY_CONTAINER_FOLDER_PATH,
                      script=SCRIPT_FILE_TO_EXECUTE,
                      arguments=EXECUTE_ARGUMENTS,
                      compute_target=compute_target,
                      environment=vm_env,
                      docker_runtime_config=docker_config)

Answer 2

为使用RunConfiguration的任何人添加另一个示例：

更改：

run_config = RunConfiguration()
run_config.environment.docker.enabled = True

至：

run_config = RunConfiguration()
docker_config = DockerConfiguration(use_docker=True)
run_config.docker = docker_config

run_config以后可以用作例如PythonScriptStep的参数。

注意，docker属性直接从内部environment移动到RunConfiguration。