From 7531bb866902aedd3095f8da9ed87a2ae0676d2f Mon Sep 17 00:00:00 2001 From: Charlie Marsh Date: Thu, 30 Jan 2025 14:05:13 -0500 Subject: [PATCH] Upgrade PyTorch version in documentation (#11114) ## Summary PyTorch 2.6.0 is now GA! --- docs/guides/integration/pytorch.md | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/docs/guides/integration/pytorch.md b/docs/guides/integration/pytorch.md index ce1bfc387..4113bc066 100644 --- a/docs/guides/integration/pytorch.md +++ b/docs/guides/integration/pytorch.md @@ -46,8 +46,8 @@ name = "project" version = "0.1.0" requires-python = ">=3.12" dependencies = [ - "torch>=2.5.1", - "torchvision>=0.20.1", + "torch>=2.6.0", + "torchvision>=0.21.0", ] ``` @@ -228,8 +228,8 @@ name = "project" version = "0.1.0" requires-python = ">=3.12.0" dependencies = [ - "torch>=2.5.1", - "torchvision>=0.20.1", + "torch>=2.6.0", + "torchvision>=0.21.0", ] [tool.uv.sources] @@ -261,8 +261,8 @@ name = "project" version = "0.1.0" requires-python = ">=3.12.0" dependencies = [ - "torch>=2.5.1", - "torchvision>=0.20.1", + "torch>=2.6.0", + "torchvision>=0.21.0", ] [tool.uv.sources] @@ -295,8 +295,8 @@ name = "project" version = "0.1.0" requires-python = ">=3.12.0" dependencies = [ - "torch>=2.5.1", - "torchvision>=0.20.1", + "torch>=2.6.0", + "torchvision>=0.21.0", "pytorch-triton-xpu>=3.2.0 ; sys_platform == 'linux'", ] @@ -336,12 +336,12 @@ dependencies = [] [project.optional-dependencies] cpu = [ - "torch>=2.5.1", - "torchvision>=0.20.1", + "torch>=2.6.0", + "torchvision>=0.21.0", ] cu124 = [ - "torch>=2.5.1", - "torchvision>=0.20.1", + "torch>=2.6.0", + "torchvision>=0.21.0", ] [tool.uv]