nvflare.app_common.resource_consumers.gpu_resource_consumer module

class GPUResourceConsumer(gpu_resource_key='gpu')[source]

Bases: ResourceConsumerSpec

consume(resources: dict)[source]