torch>=2.0
pynvml>=11.0
