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