nvflare.fuel.utils.gpu_utils module

get_host_gpu_ids() List[source]

Gets GPU IDs.

Note

Only supports nvidia-smi now.

get_host_gpu_memory_free(unit='MiB') List[source]
get_host_gpu_memory_total(unit='MiB') List[source]
has_nvidia_smi() bool[source]
use_nvidia_smi(query: str, report_format: str = 'csv')[source]