nemo_run.slurm#
Module Contents#
Classes#
Data#
API#
- class nemo_run.slurm.SlurmJobConfig#
- container_entrypoint: str#
None
- cpu_worker_memory_limit: str#
‘0’
- cudf_spill: str#
‘1’
- device: str#
‘cpu’
- interface: str#
‘eth0’
- job_dir: str#
None
- libcudf_cufile_policy: str#
‘OFF’
- protocol: str#
‘tcp’
- rapids_no_initialize: str#
‘1’
- rmm_scheduler_pool_size: str#
‘1GB’
- rmm_worker_pool_size: str#
‘72GiB’
- script_command: str#
None
- to_script(
- add_scheduler_file: bool = True,
- add_device: bool = True,
- nemo_run.slurm.run#
‘safe_import(…)’