
[dev]
pytest
torch>=2.0

[full]
torch>=2.0
pynvml>=11.0

[torch]
torch>=2.0
