Skip to content

Commit 6aa6020

Browse files
authored
[Misc] Specify minimum pynvml version (#9827)
Signed-off-by: Jee Jee Li <pandaleefree@gmail.com>
1 parent ff5ed6e commit 6aa6020

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

requirements-cuda.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44
# Dependencies for NVIDIA GPUs
55
ray >= 2.9
6-
nvidia-ml-py # for pynvml package
6+
nvidia-ml-py >= 12.560.30 # for pynvml package
77
torch == 2.5.0
88
# These must be updated alongside torch
99
torchvision == 0.20 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version

0 commit comments

Comments
 (0)