pynvml
torch
