torchtnt.utils.device.get_nvidia_smi_gpu_stats

torchtnt.utils.device.get_nvidia_smi_gpu_stats(device: device) → GPUStats

Get GPU stats from nvidia smi.

Parameters:device – A GPU torch.device to get stats from. Returns:a dict that maps gpu stats to their values.

Keys:

• ’utilization_gpu_percent’
• ’utilization_memory_percent’
• ’fan_speed_percent’
• ’memory_used_mb’
• ’memory_free_mb’
• ’temperature_gpu_celsius’
• ’temperature_memory_celsius’

Return type:dict (str, float) Raises:FileNotFoundError – If nvidia-smi command is not found.