From b7290a60487ad311b913071d46d1fde12e68fd49 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 Signed-off-by: Maxime Fournioux <55544262+mfournioux@users.noreply.github.com> --- 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