torch
numpy
scipy
matplotlib
pynvml
