dask_jobqueue.PBSCluster

dask_jobqueue.PBSCluster

class dask_jobqueue.PBSCluster(n_workers=0, job_cls: typing.Optional[dask_jobqueue.core.Job] = None, loop=None, security=None, shared_temp_directory=None, silence_logs='error', name=None, asynchronous=False, dashboard_address=None, host=None, scheduler_options=None, scheduler_cls=<class 'distributed.scheduler.Scheduler'>, interface=None, protocol=None, config_name=None, **job_kwargs)

在 PBS 集群上启动 Dask

参数
queuestr

每个工作者作业的目标队列。传递给 #PBS -q 选项。

projectstr

已弃用:请改用 account。此参数将在未来版本中移除。

accountstr

与每个工作者作业关联的记账字符串。传递给 #PBS -A 选项。

coresint

一个作业中所有工作者线程运行的 CPU 核心总数。每个工作者进程的线程数使用公式 cores / processes 确定。作业队列系统默认将其用作每个作业的 CPU 数量。

memory: str

一个作业中所有工作者使用的内存总量。作业队列系统默认将其用作每个作业的内存量。

processesint

将作业分成这么多进程。适用于 GIL 工作负载或具有许多核心的节点。默认情况下,process ~= sqrt(cores),以便进程数和每个进程的线程数大致相同。

interfacestr

网络接口,如 ‘eth0’ 或 ‘ib0’。这将用于 Dask 调度器和 Dask 工作者接口。如果您需要为 Dask 调度器使用不同的接口,可以通过 scheduler_options 参数传递:interface=your_worker_interface, scheduler_options={'interface': your_scheduler_interface}

nannybool

是否启动 nanny 进程

local_directorystr

用于文件溢出的 Dask 工作者本地目录。

death_timeoutfloat

在关闭工作者之前等待调度器的秒数

extralist

已弃用:请改用 worker_extra_args。此参数将在未来版本中移除。

worker_commandlist

启动工作者时运行的命令。默认为 “distributed.cli.dask_worker”

worker_extra_argslist

传递给 dask-worker 的额外参数

env_extralist

已弃用:请改用 job_script_prologue。此参数将在未来版本中移除。

job_script_prologuelist

在启动工作者之前添加到脚本的其他命令。

job_script_epiloguelist

工作者命令退出后添加到脚本中运行的命令。

header_skiplist

已弃用:请改用 job_directives_skip。此参数将在未来版本中移除。

job_directives_skiplist

在生成的作业脚本头部中要跳过的指令。包含指定字符串的指令行将被移除。通过 job_extra_directives 添加的指令不受影响。

log_directorystr

用于作业调度器日志的目录。

shebangstr

批处理提交脚本所需解释器的路径。

pythonstr

用于启动 Dask 工作者的 Python 可执行文件。默认为提交这些作业的 Python。

config_namestr

jobqueue.yaml 配置文件中要使用的部分。

namestr

Dask 工作者的名称。这通常由 Cluster 设置。

n_workersint

默认启动的工作者数量。默认为 0。参见 scale 方法。

silence_logsstr

如果调度器在本地启动,在此处发出的日志级别,如“debug”、“info”或“error”。

asynchronousbool

是否使用 async/await 语法运行此集群对象。

securitySecurity 或 Bool

如果您使用 TLS/SSL,则为 dask.distributed 安全对象。如果为 True,将自动创建临时的自签名凭证。

scheduler_optionsdict

用于向 Dask 调度器传递额外参数。例如,使用 scheduler_options={'dashboard_address': ':12435'} 指定 Web 仪表板应使用的端口,或使用 scheduler_options={'host': 'your-host'} 指定 Dask 调度器应运行的主机。更多详细信息请参见 distributed.Scheduler

scheduler_clstype

更改所使用的 Dask 调度器类。默认为 Dask 的 distributed.Scheduler

shared_temp_directorystr

调度器和工作者之间的共享目录(例如用于临时安全证书),如果未设置,则默认为当前工作目录。

resource_specstr

请求资源并指定作业放置。传递给 #PBS -l 选项。

walltimestr

每个工作者作业的墙钟时间。

job_extralist

已弃用:请改用 job_extra_directives。此参数将在未来版本中移除。

job_extra_directiveslist

其他 PBS 选项列表。每个选项前面都会加上 #PBS 前缀。

示例

>>> from dask_jobqueue import PBSCluster
>>> cluster = PBSCluster(queue='regular', account="myaccountingstring", cores=24,
...     memory="500 GB")
>>> cluster.scale(jobs=10)  # ask for 10 jobs
>>> from dask.distributed import Client
>>> client = Client(cluster)

这也适用于自适应集群。它根据负载自动启动和终止工作者。

>>> cluster.adapt(maximum_jobs=20)
__init__(n_workers=0, job_cls: typing.Optional[dask_jobqueue.core.Job] = None, loop=None, security=None, shared_temp_directory=None, silence_logs='error', name=None, asynchronous=False, dashboard_address=None, host=None, scheduler_options=None, scheduler_cls=<class 'distributed.scheduler.Scheduler'>, interface=None, protocol=None, config_name=None, **job_kwargs)

方法

__init__([n_workers, job_cls, loop, ...])

adapt(*args[, minimum_jobs, maximum_jobs])

根据调度器活动自动伸缩 Dask 集群。

close([timeout])

from_name(name)

通过名称创建此类的实例来表示现有集群。

get_client()

返回集群的客户端

get_logs([cluster, scheduler, workers])

返回集群、调度器和工作者的日志

job_script()

logs(*args, **kwargs)

new_worker_spec()

返回下一个工作者的名称和规格

scale([n, jobs, memory, cores])

将集群伸缩到指定配置。

scale_down(workers)

scale_up([n, memory, cores])

将集群伸缩到 n 个工作者

sync(func, *args[, asynchronous, ...])

根据调用上下文同步或异步调用 func 并传递 args

wait_for_workers(n_workers[, timeout])

阻塞调用,等待 n 个工作者准备好后继续

属性

asynchronous

我们是否在事件循环中运行?

called_from_running_loop

dashboard_link

job_header

job_name

loop

name

observed

plan

requested

scheduler_address