serverless_gpu.compute

GPU compute type definitions and utilities.

This module defines the available GPU types for serverless compute and provides utilities for working with GPU configurations and resource allocation.

Functions

Classes

GPUType(value)

Enumeration of available GPU types for serverless compute.

class serverless_gpu.compute.GPUType(value)[source]

Bases: Enum

Enumeration of available GPU types for serverless compute.

This enum defines the GPU types that can be used for distributed computing on the serverless GPU platform.

H100: NVIDIA H100 80GB GPU instances.

A10: NVIDIA A10 GPU instances.

A10 = 'a10'

H100 = 'h100_80gb'

classmethod values()[source]

Get all GPU type values.

Returns: List of all available GPU type values.
Return type: List[str]