Slurm launcher in Python.
Project description
slrm
Python library to submit slurm jobs.
Installation:
pip install git+https://github.com/vzhong/slrm
Usage:
import os
import slrm
for city, state in [('vancouver', 'bc'), ('toronto', 'on'), ('stanford', 'ca'), ('seattle', 'wa')]:
cmd = 'python myscript.py --city {} --state {}'.format(city, state)
slurm_kwargs = {'account': 'cse', 'partition': 'cse-gpu'}
# you can see default args in slrm.launcher.default_args
slrm.launch(cmd, slurm_kwargs, dout=os.path.join('jobs', '{}-{}'.format(city, state)), dry=False)
Pull requests welcome!
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
slrm-0.0.1.tar.gz
(3.4 kB
view hashes)
Built Distribution
slrm-0.0.1-py3-none-any.whl
(3.5 kB
view hashes)