ivis.models.utils.gpu

Functions

print_gpu_memory([device])

Print GPU memory usage (allocated, reserved, total) for the specified device.

ivis.models.utils.gpu.print_gpu_memory(device='cuda:0')

Print GPU memory usage (allocated, reserved, total) for the specified device.

Parameters:

device (str or torch.device) – CUDA device string, e.g., “cuda:0”.