Skip to content

GitLab

Unverified Commit 6aa6020f authored by Jee Jee Li's avatar Jee Jee Li Committed by GitHub
Browse files

[Misc] Specify minimum pynvml version (#9827) 


Signed-off-by: default avatarJee Jee Li <pandaleefree@gmail.com>
parent ff5ed6e1
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
requirements-cuda.txt
View file @ 6aa6020f
...@@ -3,7 +3,7 @@ ...@@ -3,7 +3,7 @@
# Dependencies for NVIDIA GPUs # Dependencies for NVIDIA GPUs
ray >= 2.9 ray >= 2.9
nvidia-ml-py # for pynvml package nvidia-ml-py >= 12.560.30 # for pynvml package
torch == 2.5.0 torch == 2.5.0
# These must be updated alongside torch # 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 torchvision == 0.20 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment