stages.resources#
Module Contents#
Classes#
Define resource requirements for a processing stage. |
API#
- class stages.resources.Resources#
Define resource requirements for a processing stage.
Attributes: cpus: Number of CPU cores required gpu_memory_gb: GPU memory required in GB (Only for single-GPU stages) nvdecs: Number of NVDEC units required nvencs: Number of NVENC units required entire_gpu: Whether to allocate entire GPU regardless of memory (This also gives you nvdecs and nvencs of that GPU) gpus: Number of GPUs required (Only for multi-GPU stages)
- cpus: float#
1.0
- entire_gpu: bool#
False
- gpu_memory_gb: float#
0.0
- gpus: float#
0.0
- nvdecs: int#
0
- nvencs: int#
0
- property requires_gpu: bool#
Check if this stage requires GPU resources.