diff --git a/pyproject.toml b/pyproject.toml index c9057b061aad..c5368c9f815e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -6,7 +6,7 @@ requires = [ "packaging", "setuptools>=61", "setuptools-scm>=8.0", - "torch == 2.4.0", + "torch == 2.6.0", "wheel", "jinja2", ] diff --git a/requirements-build.txt b/requirements-build.txt index 6144a56da8c4..3248302b3dd7 100644 --- a/requirements-build.txt +++ b/requirements-build.txt @@ -4,6 +4,6 @@ ninja packaging setuptools>=61 setuptools-scm>=8 -torch==2.4.0 +torch==2.6.0 wheel jinja2 diff --git a/requirements-cpu.txt b/requirements-cpu.txt index 27ca8ca5dbc5..14f3274843fd 100644 --- a/requirements-cpu.txt +++ b/requirements-cpu.txt @@ -2,5 +2,5 @@ -r requirements-common.txt # Dependencies for x86_64 CPUs -torch == 2.4.0+cpu; platform_machine != "ppc64le" +torch == 2.6.0; platform_machine != "ppc64le" torchvision; platform_machine != "ppc64le" # required for the image processor of phi3v, this must be updated alongside torch diff --git a/requirements-cuda.txt b/requirements-cuda.txt index 3b3c2f876919..5da3ca64a4f1 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.txt @@ -4,7 +4,7 @@ # Dependencies for NVIDIA GPUs ray >= 2.9 nvidia-ml-py # for pynvml package -torch == 2.4.0 +torch == 2.6.0 # These must be updated alongside torch torchvision == 0.19 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version xformers == 0.0.27.post2; platform_system == 'Linux' and platform_machine == 'x86_64' # Requires PyTorch 2.4.0 diff --git a/requirements-xpu.txt b/requirements-xpu.txt index ce83a178c618..bfc05b0408d3 100644 --- a/requirements-xpu.txt +++ b/requirements-xpu.txt @@ -9,7 +9,7 @@ setuptools-scm>=8 wheel jinja2 # Following pkgs retrieved from https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ -torch == 2.3.1+cxx11.abi +torch == 2.6.0 intel-extension-for-pytorch == 2.3.110+xpu oneccl_bind_pt == 2.3.100+xpu