validate_model_on_gpu.py

import nvgpu

gpu_info = nvgpu.gpu_info()

model_loaded = False

for info in gpu_info:
    if info['mem_used'] > 0 and info['mem_used_percent'] > 0.0:
        model_loaded = True
        break

if not model_loaded:
    exit(1)