跳转到主要内容

Python的SLURM工具

项目描述

pySlurmUtils

从Python调度作业的SLURM工具。

该库的主要目的是提供一个适用于SLURM的concurrent.futures.Executor实现,可用于任何机器,而不仅仅是SLURM客户端。

pip install pyslurmutils
from pyslurmutils.concurrent.futures import SlurmRestExecutor

with SlurmRestExecutor(
    url=url,                        # SLURM REST URL
    user_name=user_name,            # SLURM user name
    token=token,                    # SLURM access token
    log_directory="/path/to/log",   # for log files (optional)
    data_directory="/path/to/data", # TCP communication when not provided
    pre_script="module load ewoks", # load environment (optional)
    parameters={"time_limit": 120}  # SLURM job parameters (optional)
    python_cmd="python",            # python command (python3 by default)
) as executor:
    future = executor.submit(sum, [1, 1])
    assert future.result() == 2

文档

https://pyslurmutils.readthedocs.io/

项目详情


下载文件

下载适用于您平台的文件。如果您不确定选择哪个,请了解有关安装包的更多信息。

源分发

pyslurmutils-0.2.0.tar.gz (29.9 kB 查看哈希值)

上传时间

支持者