From e4bf860a54a302ccb2d80489368d5df686e46923 Mon Sep 17 00:00:00 2001 From: youkaichao Date: Tue, 23 Apr 2024 18:33:12 -0700 Subject: [PATCH] [CI][Build] change pynvml to nvidia-ml-py (#4302) --- requirements-cuda.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements-cuda.txt b/requirements-cuda.txt index c6d2cd46..1bddae4c 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.txt @@ -3,7 +3,7 @@ # Dependencies for NVIDIA GPUs ray >= 2.9 -pynvml == 11.5.0 +nvidia-ml-py # for pynvml package vllm-nccl-cu12>=2.18,<2.19 # for downloading nccl library torch == 2.2.1 xformers == 0.0.25 # Requires PyTorch 2.2.1