Skip to content

Commit aa0928b

Browse files
authored
Add 'cuda_bindings' to dependencies list (#7483)
1 parent 89981f3 commit aa0928b

File tree

2 files changed

+27
-1
lines changed

2 files changed

+27
-1
lines changed

s3_management/manage.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -132,6 +132,7 @@
132132
"charset_normalizer",
133133
"cmake",
134134
"colorama",
135+
"cuda_bindings",
135136
"fbgemm_gpu",
136137
"fbgemm_gpu_genai",
137138
"filelock",

s3_management/update_dependencies.py

Lines changed: 26 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,32 @@
1919
"fsspec": [{"version": "latest", "project": "torch"}],
2020
"nvidia-cudnn-cu11": [{"version": "latest", "project": "torch"}],
2121
"typing-extensions": [{"version": "latest", "project": "torch"}],
22+
"cuda-bindings": [
23+
{
24+
"version": "12.9.4",
25+
"project": "torch",
26+
"target": "cu126",
27+
},
28+
{
29+
"version": "12.9.4",
30+
"project": "torch",
31+
"target": "cu128",
32+
},
33+
{
34+
"version": "12.9.4",
35+
"project": "torch",
36+
"target": "cu129",
37+
},
38+
{
39+
"version": "13.0.3",
40+
"project": "torch",
41+
"target": "cu130",
42+
},
43+
{
44+
"version": "12.9.2",
45+
"project": "vllm",
46+
},
47+
],
2248
"nvidia-cuda-nvrtc-cu12": [
2349
{
2450
"version": "12.6.77",
@@ -451,7 +477,6 @@
451477
# vLLM
452478
"ninja": [{"version": "latest", "project": "vllm"}],
453479
"cuda-python": [{"version": "12.9.0", "project": "vllm"}],
454-
"cuda-bindings": [{"version": "12.9.2", "project": "vllm"}],
455480
"cuda-pathfinder": [{"version": "latest", "project": "vllm"}],
456481
"pynvml": [{"version": "latest", "project": "vllm"}],
457482
"nvidia-ml-py": [{"version": "latest", "project": "vllm"}],

0 commit comments

Comments
 (0)