Skip to content

Commit 3228e5d

Browse files
22quinnmawong-amd
authored andcommitted
Update test requirements to CUDA 12.8 (vllm-project#17576)
Signed-off-by: 22quinn <33176974+22quinn@users.noreply.github.com>
1 parent a3de12a commit 3228e5d

File tree

2 files changed

+17
-17
lines changed

2 files changed

+17
-17
lines changed

.pre-commit-config.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ repos:
4646
rev: 0.6.17
4747
hooks:
4848
- id: pip-compile
49-
args: [requirements/test.in, -o, requirements/test.txt, --index-strategy, unsafe-best-match]
49+
args: [requirements/test.in, -o, requirements/test.txt, --index-strategy, unsafe-best-match, --torch-backend, cu128]
5050
files: ^requirements/test\.(in|txt)$
5151
- repo: local
5252
hooks:

requirements/test.txt

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# This file was autogenerated by uv via the following command:
2-
# uv pip compile requirements/test.in -o requirements/test.txt --index-strategy unsafe-best-match
2+
# uv pip compile requirements/test.in -o requirements/test.txt --index-strategy unsafe-best-match --torch-backend cu128
33
absl-py==2.1.0
44
# via rouge-score
55
accelerate==1.0.1
@@ -349,42 +349,42 @@ numpy==1.26.4
349349
# transformers
350350
# tritonclient
351351
# vocos
352-
nvidia-cublas-cu12==12.6.4.1
352+
nvidia-cublas-cu12==12.8.3.14
353353
# via
354354
# nvidia-cudnn-cu12
355355
# nvidia-cusolver-cu12
356356
# torch
357-
nvidia-cuda-cupti-cu12==12.6.80
357+
nvidia-cuda-cupti-cu12==12.8.57
358358
# via torch
359-
nvidia-cuda-nvrtc-cu12==12.6.77
359+
nvidia-cuda-nvrtc-cu12==12.8.61
360360
# via torch
361-
nvidia-cuda-runtime-cu12==12.6.77
361+
nvidia-cuda-runtime-cu12==12.8.57
362362
# via torch
363-
nvidia-cudnn-cu12==9.5.1.17
363+
nvidia-cudnn-cu12==9.7.1.26
364364
# via torch
365-
nvidia-cufft-cu12==11.3.0.4
365+
nvidia-cufft-cu12==11.3.3.41
366366
# via torch
367-
nvidia-cufile-cu12==1.11.1.6
367+
nvidia-cufile-cu12==1.13.0.11
368368
# via torch
369-
nvidia-curand-cu12==10.3.7.77
369+
nvidia-curand-cu12==10.3.9.55
370370
# via torch
371-
nvidia-cusolver-cu12==11.7.1.2
371+
nvidia-cusolver-cu12==11.7.2.55
372372
# via torch
373-
nvidia-cusparse-cu12==12.5.4.2
373+
nvidia-cusparse-cu12==12.5.7.53
374374
# via
375375
# nvidia-cusolver-cu12
376376
# torch
377377
nvidia-cusparselt-cu12==0.6.3
378378
# via torch
379379
nvidia-nccl-cu12==2.26.2
380380
# via torch
381-
nvidia-nvjitlink-cu12==12.6.85
381+
nvidia-nvjitlink-cu12==12.8.61
382382
# via
383383
# nvidia-cufft-cu12
384384
# nvidia-cusolver-cu12
385385
# nvidia-cusparse-cu12
386386
# torch
387-
nvidia-nvtx-cu12==12.6.77
387+
nvidia-nvtx-cu12==12.8.55
388388
# via torch
389389
opencv-python-headless==4.11.0.86
390390
# via
@@ -687,7 +687,7 @@ tomli==2.2.1
687687
# via schemathesis
688688
tomli-w==1.2.0
689689
# via schemathesis
690-
torch==2.7.0
690+
torch==2.7.0+cu128
691691
# via
692692
# -r requirements/test.in
693693
# accelerate
@@ -705,12 +705,12 @@ torch==2.7.0
705705
# torchvision
706706
# vector-quantize-pytorch
707707
# vocos
708-
torchaudio==2.7.0
708+
torchaudio==2.7.0+cu128
709709
# via
710710
# -r requirements/test.in
711711
# encodec
712712
# vocos
713-
torchvision==0.22.0
713+
torchvision==0.22.0+cu128
714714
# via
715715
# -r requirements/test.in
716716
# timm

0 commit comments

Comments
 (0)