WorkerConfiguration 类

WorkerConfiguration 是保存辅助角色运行所需全部信息的类。

继承

azureml._base_sdk_common.abstract_run_config_element._AbstractRunConfigElement

WorkerConfiguration

构造函数

WorkerConfiguration(node_count, compute_target=None, environment=None, shm_size='2g', history=None, use_gpu=False, pip_packages=None, conda_packages=None, conda_dependencies_file=None, pip_requirements_file=None)

参数

node_count

int

必需

要初始化的工作器节点数，将在计算目标中的每台计算机上运行一个辅助角色。

compute_target

AbstractComputeTarget 或 str

默认值: None

将在其中运行辅助角色的计算目标。 这可以是一个对象或计算目标的名称。

environment

Environment

默认值: None

辅助角色的环境定义。 它包括 PythonSection、DockerSection 和环境变量。 使用此参数可以设置不通过其他参数直接公开到 WorkerConfiguration 构造的任何环境选项。 如果指定此参数，将基于此参数添加 pip_packages 和 conda_packages 中指定的包。

shm_size

str

默认值: 2g

辅助角色的 Docker shm_size 配置。

history

HistoryConfiguration

默认值: None

辅助角色的运行的历史记录配置，它对将要监视哪些日志文件夹进行控制

use_gpu

bool

默认值: False

参数，用于指示是否应为默认的基础映像添加 gpu 包。 如果 environment 已设置，则忽略此参数。

conda_packages

list

默认值: None

一个字符串列表，表示要添加到此辅助角色的 Python 环境中的 conda 包。

pip_packages

list

默认值: None

一个字符串列表，表示要添加到辅助角色的 Python 环境中的 pip 包。

pip_requirements_file

str

默认值: None

辅助角色的 pip 要求文本文件的相对路径。 这可以与 pip_packages 参数一起提供。

conda_dependencies_file

str

默认值: None

辅助角色的 conda 依赖项 yaml 文件的相对路径。

属性

target

获取计划在其中执行辅助角色运行的计算目标。

可以使用函数 compute_targets 找到可用的云计算目标

返回

目标名称

返回类型

str