diff --git a/pyproject.toml b/pyproject.toml index 2a00d6796ee..a87736e5425 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -5,7 +5,7 @@ requires = [ "ninja", "packaging", "setuptools >= 49.4.0", - "torch == 2.2.1", + "torch == 2.7.1", "wheel", ] build-backend = "setuptools.build_meta" diff --git a/requirements-build.txt b/requirements-build.txt index 2bc07fb152a..3a594aae28f 100644 --- a/requirements-build.txt +++ b/requirements-build.txt @@ -3,5 +3,5 @@ cmake>=3.21 ninja packaging setuptools>=49.4.0 -torch==2.2.1 +torch==2.7.1 wheel diff --git a/requirements-cpu.txt b/requirements-cpu.txt index 36d20bc9473..e7f98edd001 100644 --- a/requirements-cpu.txt +++ b/requirements-cpu.txt @@ -2,5 +2,5 @@ -r requirements-common.txt # Dependencies for x86_64 CPUs -torch == 2.2.1+cpu +torch == 2.7.1 triton >= 2.1.0 # FIXME(woosuk): This is a hack to avoid import error. diff --git a/requirements-cuda.txt b/requirements-cuda.txt index 6ee75e8139c..3be2d919b7f 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.txt @@ -5,6 +5,6 @@ ray >= 2.9 pynvml == 11.5.0 vllm-nccl-cu12>=2.18,<2.19 # for downloading nccl library -torch == 2.2.1 +torch == 2.7.1 xformers == 0.0.25 # Requires PyTorch 2.2.1 triton >= 2.1.0