nvidia-ml-py>=12
psutil>=5.9

[all]
torch>=2.0

[hooks]
torch>=2.0
