cuperiod.GpuInfo

class cuperiod.GpuInfo(device_id, name, total_bytes, free_bytes, mps_enabled)[source]

Bases: object

A snapshot of one CUDA device’s identity and free memory.

Parameters:
  • device_id (int)

  • name (str)

  • total_bytes (int)

  • free_bytes (int)

  • mps_enabled (bool)

property total_gib: float

Total device memory in GiB.

property free_gib: float

Free device memory in GiB.