xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-10 08:25:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update test requirements to CUDA 12.8 (#17576)

Browse Source 
Signed-off-by: 22quinn <33176974+22quinn@users.noreply.github.com>
This commit is contained in:
22quinn 2025-05-02 21:40:15 -07:00 committed by GitHub
parent 22c6f6397f
commit d47b605eca
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.pre-commit-config.yaml
View File

@ -46,7 +46,7 @@ repos:
rev: 0.6.17 rev: 0.6.17
hooks: hooks:
- id: pip-compile - id: pip-compile
args: [requirements/test.in, -o, requirements/test.txt, --index-strategy, unsafe-best-match] args: [requirements/test.in, -o, requirements/test.txt, --index-strategy, unsafe-best-match, --torch-backend, cu128]
files: ^requirements/test\.(in|txt)$ files: ^requirements/test\.(in|txt)$
- repo: local - repo: local
hooks: hooks:

32
requirements/test.txt
View File

@ -1,5 +1,5 @@
# This file was autogenerated by uv via the following command: # This file was autogenerated by uv via the following command:
# uv pip compile requirements/test.in -o requirements/test.txt --index-strategy unsafe-best-match # uv pip compile requirements/test.in -o requirements/test.txt --index-strategy unsafe-best-match --torch-backend cu128
absl-py==2.1.0 absl-py==2.1.0
# via rouge-score # via rouge-score
accelerate==1.0.1 accelerate==1.0.1
@ -349,28 +349,28 @@ numpy==1.26.4
# transformers # transformers
# tritonclient # tritonclient
# vocos # vocos
nvidia-cublas-cu12==12.6.4.1 nvidia-cublas-cu12==12.8.3.14
# via # via
# nvidia-cudnn-cu12 # nvidia-cudnn-cu12
# nvidia-cusolver-cu12 # nvidia-cusolver-cu12
# torch # torch
nvidia-cuda-cupti-cu12==12.6.80 nvidia-cuda-cupti-cu12==12.8.57
# via torch # via torch
nvidia-cuda-nvrtc-cu12==12.6.77 nvidia-cuda-nvrtc-cu12==12.8.61
# via torch # via torch
nvidia-cuda-runtime-cu12==12.6.77 nvidia-cuda-runtime-cu12==12.8.57
# via torch # via torch
nvidia-cudnn-cu12==9.5.1.17 nvidia-cudnn-cu12==9.7.1.26
# via torch # via torch
nvidia-cufft-cu12==11.3.0.4 nvidia-cufft-cu12==11.3.3.41
# via torch # via torch
nvidia-cufile-cu12==1.11.1.6 nvidia-cufile-cu12==1.13.0.11
# via torch # via torch
nvidia-curand-cu12==10.3.7.77 nvidia-curand-cu12==10.3.9.55
# via torch # via torch
nvidia-cusolver-cu12==11.7.1.2 nvidia-cusolver-cu12==11.7.2.55
# via torch # via torch
nvidia-cusparse-cu12==12.5.4.2 nvidia-cusparse-cu12==12.5.7.53
# via # via
# nvidia-cusolver-cu12 # nvidia-cusolver-cu12
# torch # torch
@ -378,13 +378,13 @@ nvidia-cusparselt-cu12==0.6.3
# via torch # via torch
nvidia-nccl-cu12==2.26.2 nvidia-nccl-cu12==2.26.2
# via torch # via torch
nvidia-nvjitlink-cu12==12.6.85 nvidia-nvjitlink-cu12==12.8.61
# via # via
# nvidia-cufft-cu12 # nvidia-cufft-cu12
# nvidia-cusolver-cu12 # nvidia-cusolver-cu12
# nvidia-cusparse-cu12 # nvidia-cusparse-cu12
# torch # torch
nvidia-nvtx-cu12==12.6.77 nvidia-nvtx-cu12==12.8.55
# via torch # via torch
opencv-python-headless==4.11.0.86 opencv-python-headless==4.11.0.86
# via # via
@ -687,7 +687,7 @@ tomli==2.2.1
# via schemathesis # via schemathesis
tomli-w==1.2.0 tomli-w==1.2.0
# via schemathesis # via schemathesis
torch==2.7.0 torch==2.7.0+cu128
# via # via
# -r requirements/test.in # -r requirements/test.in
# accelerate # accelerate
@ -705,12 +705,12 @@ torch==2.7.0
# torchvision # torchvision
# vector-quantize-pytorch # vector-quantize-pytorch
# vocos # vocos
torchaudio==2.7.0 torchaudio==2.7.0+cu128
# via # via
# -r requirements/test.in # -r requirements/test.in
# encodec # encodec
# vocos # vocos
torchvision==0.22.0 torchvision==0.22.0+cu128
# via # via
# -r requirements/test.in # -r requirements/test.in
# timm # timm