Skip to content

Commit e4bf860

Browse files
authored
[CI][Build] change pynvml to nvidia-ml-py (#4302)
1 parent 91f50a6 commit e4bf860

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-
pynvml == 11.5.0
6+
nvidia-ml-py # for pynvml package
77
vllm-nccl-cu12>=2.18,<2.19 # for downloading nccl library
88
torch == 2.2.1
99
xformers == 0.0.25 # Requires PyTorch 2.2.1

0 commit comments

Comments
 (0)