rockverse.configure.Config.rank_select_gpu#

Config.rank_select_gpu()[source]#

Selects the GPU to be used by the current MPI process at runtime. The selection is based on the user-select GPUs.

Returns:

The index of the selected GPU, or None if no GPUs are available.