sporco_cuda.util module¶
-
device_count()¶ Get the number of CUDA GPU devices installed on the host system.
- Returns
- ndevint
Number of installed deviced
-
current_device(id=None)¶ Get or set the current CUDA GPU device. The current device is not set if id is None
- Parameters
- idint or None, optional (default None)
Device number of device to be set as current device
- Returns
- idint
Device number of current device
-
memory_info()¶ Get memory information for the current CUDA GPU device.
- Returns
- freeint
Free memory in bytes
- totalint
Total memory in bytes
-
device_name(int dev=0)¶ Get hardware model name for the specified CUDA GPU device.
- Parameters
- idint, optional (default 0)
Device number of device
- Returns
- namestring
Hardware device name