gpustat
torchvision
