def test_get_nvidia_gpu_stats(tmpdir):
"""Test GPU get_device_stats with Pytorch < 1.8.0."""
current_device = torch.device(f"cuda:{torch.cuda.current_device()}")
gpu_stats = GPUAccelerator().get_device_stats(current_device)
fields = [
"utilization.gpu", "memory.used", "memory.free", "utilization.memory"
]
for f in fields:
assert any(f in h for h in gpu_stats.keys())
def test_get_torch_gpu_stats(tmpdir):
current_device = torch.device(f"cuda:{torch.cuda.current_device()}")
gpu_stats = GPUAccelerator().get_device_stats(current_device)
fields = [
"allocated_bytes.all.freed", "inactive_split.all.peak",
"reserved_bytes.large_pool.peak"
]
for f in fields:
assert any(f in h for h in gpu_stats.keys())