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

Update PyTorch to torch==2.7.1 for CUDA (#21011)

Browse Source 
Signed-off-by: mgoin <mgoin64@gmail.com>
This commit is contained in:
Michael Goin 2025-07-16 22:30:44 -04:00 committed by GitHub
parent 72ad273582
commit 4e7dfbe7b4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
7 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
CMakeLists.txt
View File

@ -45,7 +45,7 @@ set(HIP_SUPPORTED_ARCHS "gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1
# requirements.txt files and should be kept consistent. The ROCm torch # requirements.txt files and should be kept consistent. The ROCm torch
# versions are derived from docker/Dockerfile.rocm # versions are derived from docker/Dockerfile.rocm
# #
set(TORCH_SUPPORTED_VERSION_CUDA "2.7.0") set(TORCH_SUPPORTED_VERSION_CUDA "2.7.1")
set(TORCH_SUPPORTED_VERSION_ROCM "2.7.0") set(TORCH_SUPPORTED_VERSION_ROCM "2.7.0")
# #

2
pyproject.toml
View File

@ -6,7 +6,7 @@ requires = [
"packaging>=24.2", "packaging>=24.2",
"setuptools>=77.0.3,<80.0.0", "setuptools>=77.0.3,<80.0.0",
"setuptools-scm>=8.0", "setuptools-scm>=8.0",
"torch == 2.7.0", "torch == 2.7.1",
"wheel", "wheel",
"jinja2", "jinja2",
] ]

2
requirements/build.txt
View File

@ -4,7 +4,7 @@ ninja
packaging>=24.2 packaging>=24.2
setuptools>=77.0.3,<80.0.0 setuptools>=77.0.3,<80.0.0
setuptools-scm>=8 setuptools-scm>=8
torch==2.7.0 torch==2.7.1
wheel wheel
jinja2>=3.1.6 jinja2>=3.1.6
regex regex

10
requirements/cuda.txt
View File

@ -6,9 +6,9 @@ numba == 0.61.2; python_version > '3.9'
# Dependencies for NVIDIA GPUs # Dependencies for NVIDIA GPUs
ray[cgraph]>=2.43.0, !=2.44.* # Ray Compiled Graph, required for pipeline parallelism in V1. ray[cgraph]>=2.43.0, !=2.44.* # Ray Compiled Graph, required for pipeline parallelism in V1.
torch==2.7.0 torch==2.7.1
torchaudio==2.7.0 torchaudio==2.7.1
# These must be updated alongside torch # These must be updated alongside torch
torchvision==0.22.0 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version torchvision==0.22.1 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version
# https://github.com/facebookresearch/xformers/releases/tag/v0.0.30 # https://github.com/facebookresearch/xformers/releases/tag/v0.0.31
xformers==0.0.30; platform_system == 'Linux' and platform_machine == 'x86_64' # Requires PyTorch >= 2.7 xformers==0.0.31; platform_system == 'Linux' and platform_machine == 'x86_64' # Requires PyTorch >= 2.7

6
requirements/test.in
View File

@ -22,9 +22,9 @@ sentence-transformers # required for embedding tests
soundfile # required for audio tests soundfile # required for audio tests
jiwer # required for audio tests jiwer # required for audio tests
timm # required for internvl test timm # required for internvl test
torch==2.7.0 torch==2.7.1
torchaudio==2.7.0 torchaudio==2.7.1
torchvision==0.22.0 torchvision==0.22.1
transformers_stream_generator # required for qwen-vl test transformers_stream_generator # required for qwen-vl test
mamba_ssm # required for plamo2 test mamba_ssm # required for plamo2 test
matplotlib # required for qwen-vl test matplotlib # required for qwen-vl test

8
requirements/test.txt
View File

@ -762,7 +762,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+cu128 torch==2.7.1+cu128
# via # via
# -r requirements/test.in # -r requirements/test.in
# accelerate # accelerate
@ -781,12 +781,12 @@ torch==2.7.0+cu128
# torchvision # torchvision
# vector-quantize-pytorch # vector-quantize-pytorch
# vocos # vocos
torchaudio==2.7.0+cu128 torchaudio==2.7.1+cu128
# via # via
# -r requirements/test.in # -r requirements/test.in
# encodec # encodec
# vocos # vocos
torchvision==0.22.0+cu128 torchvision==0.22.1+cu128
# via # via
# -r requirements/test.in # -r requirements/test.in
# timm # timm
@ -816,7 +816,7 @@ transformers==4.53.2
# transformers-stream-generator # transformers-stream-generator
transformers-stream-generator==0.0.5 transformers-stream-generator==0.0.5
# via -r requirements/test.in # via -r requirements/test.in
triton==3.3.0 triton==3.3.1
# via torch # via torch
tritonclient==2.51.0 tritonclient==2.51.0
# via # via

4
tests/entrypoints/openai/test_vision.py
View File

@ -36,11 +36,11 @@ EXPECTED_MM_BEAM_SEARCH_RES = [
], ],
[ [
"The image shows a Venn diagram with three over", "The image shows a Venn diagram with three over",
"This image shows a Venn diagram with three over", "The image shows a Venn diagram with three intersect",
], ],
[ [
"This image displays a gradient of colors ranging from", "This image displays a gradient of colors ranging from",
"This image displays a gradient of colors transitioning from", "The image displays a gradient of colors ranging from",
], ],
] ]