You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: requirements-cuda.txt
+1-1Lines changed: 1 addition & 1 deletion
Original file line number
Diff line number
Diff line change
@@ -3,7 +3,7 @@
3
3
4
4
# Dependencies for NVIDIA GPUs
5
5
ray >= 2.9
6
-
nvidia-ml-py # for pynvml package
6
+
nvidia-ml-py >= 12.560.30 # for pynvml package
7
7
torch == 2.5.0
8
8
# These must be updated alongside torch
9
9
torchvision == 0.20 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version
0 commit comments