cascade.utils.autodevice

cascade.utils.autodevice(n=1)

Get torch computation device automatically based on GPU availability and memory usage

Parameters:

n (int) – Number of GPUs to request

Return type:

tuple[str, list[int] | str]

Returns:

• Accelerator type

• List of granted devices