noether.core.distributed.run

Submodules

Functions

accelerator_to_device(accelerator)

check_single_device_visible(accelerator)

log_device_info(accelerator, device_ids)

parse_devices(accelerator, devices)

run_managed(main[, accelerator, devices])

run_unmanaged(main, devices[, accelerator, master_port])

run(main[, devices, accelerator, master_port])

Package Contents

noether.core.distributed.run.accelerator_to_device(accelerator)
noether.core.distributed.run.check_single_device_visible(accelerator)
noether.core.distributed.run.log_device_info(accelerator, device_ids)
noether.core.distributed.run.parse_devices(accelerator, devices)
Parameters:

  • accelerator (str)

  • devices (str | None)

noether.core.distributed.run.run_managed(main, accelerator='gpu', devices=None)
noether.core.distributed.run.run_unmanaged(main, devices, accelerator='gpu', master_port=None)
Parameters:

  • devices (str | None)

  • accelerator (str)

  • master_port (int | None)

noether.core.distributed.run.run(main, devices=None, accelerator='gpu', master_port=None)