From 6aa6020f9bd4c1e414c10f7bd3a7c2555f1950b2 Mon Sep 17 00:00:00 2001 From: Jee Jee Li Date: Wed, 30 Oct 2024 14:05:43 +0800 Subject: [PATCH] [Misc] Specify minimum pynvml version (#9827) Signed-off-by: Jee Jee Li --- requirements-cuda.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements-cuda.txt b/requirements-cuda.txt index 92fa303d687a2..282ab11838bf4 100644 --- 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