[Misc] Specify minimum pynvml version (vllm-project#9827)

Signed-off-by: Jee Jee Li <[email protected]>
neuralmagic · Oct 30, 2024 · 6aa6020 · 6aa6020
1 parent ff5ed6e
commit 6aa6020
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements-cuda.txt b/requirements-cuda.txt
@@ -3,7 +3,7 @@
 
 # Dependencies for NVIDIA GPUs
 ray >= 2.9
-nvidia-ml-py # for pynvml package
+nvidia-ml-py >= 12.560.30 # for pynvml package
 torch == 2.5.0
 # These must be updated alongside torch
 torchvision == 0.20   # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version