Python
/
uv
mirror of https://github.com/astral-sh/uv
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Upgrade PyTorch guide to CUDA 12.8 and PyTorch 2.7 (#13199)

This commit is contained in:
Charlie Marsh 2025-04-29 15:08:13 -04:00 committed by GitHub
parent b1fd3e3409
commit 241b013600
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 21 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
docs/guides/integration/pytorch.md
View File

@ -45,8 +45,8 @@ name = "project"
version = "0.1.0" version = "0.1.0"
requires-python = ">=3.12" requires-python = ">=3.12"
dependencies = [ dependencies = [
"torch>=2.6.0", "torch>=2.7.0",
"torchvision>=0.21.0", "torchvision>=0.22.0",
] ]
``` ```
@ -260,18 +260,18 @@ name = "project"
version = "0.1.0" version = "0.1.0"
requires-python = ">=3.12.0" requires-python = ">=3.12.0"
dependencies = [ dependencies = [
"torch>=2.6.0", "torch>=2.7.0",
"torchvision>=0.21.0", "torchvision>=0.22.0",
] ]
[tool.uv.sources] [tool.uv.sources]
torch = [ torch = [
{ index = "pytorch-cpu", marker = "sys_platform != 'linux'" }, { index = "pytorch-cpu", marker = "sys_platform != 'linux'" },
{ index = "pytorch-cu124", marker = "sys_platform == 'linux'" }, { index = "pytorch-cu128", marker = "sys_platform == 'linux'" },
] ]
torchvision = [ torchvision = [
{ index = "pytorch-cpu", marker = "sys_platform != 'linux'" }, { index = "pytorch-cpu", marker = "sys_platform != 'linux'" },
{ index = "pytorch-cu124", marker = "sys_platform == 'linux'" }, { index = "pytorch-cu128", marker = "sys_platform == 'linux'" },
] ]
[[tool.uv.index]] [[tool.uv.index]]
@ -280,8 +280,8 @@ url = "https://download.pytorch.org/whl/cpu"
explicit = true explicit = true
[[tool.uv.index]] [[tool.uv.index]]
name = "pytorch-cu124" name = "pytorch-cu128"
url = "https://download.pytorch.org/whl/cu124" url = "https://download.pytorch.org/whl/cu128"
explicit = true explicit = true
``` ```
@ -294,9 +294,9 @@ name = "project"
version = "0.1.0" version = "0.1.0"
requires-python = ">=3.12.0" requires-python = ">=3.12.0"
dependencies = [ dependencies = [
"torch>=2.6.0", "torch>=2.7.0",
"torchvision>=0.21.0", "torchvision>=0.22.0",
"pytorch-triton-xpu>=3.2.0 ; sys_platform == 'linux'", "pytorch-triton-xpu>=3.3.0 ; sys_platform == 'linux'",
] ]
[tool.uv.sources] [tool.uv.sources]
@ -335,30 +335,30 @@ dependencies = []
[project.optional-dependencies] [project.optional-dependencies]
cpu = [ cpu = [
"torch>=2.6.0", "torch>=2.7.0",
"torchvision>=0.21.0", "torchvision>=0.22.0",
] ]
cu124 = [ cu128 = [
"torch>=2.6.0", "torch>=2.7.0",
"torchvision>=0.21.0", "torchvision>=0.22.0",
] ]
[tool.uv] [tool.uv]
conflicts = [ conflicts = [
[ [
{ extra = "cpu" }, { extra = "cpu" },
{ extra = "cu124" }, { extra = "cu128" },
], ],
] ]
[tool.uv.sources] [tool.uv.sources]
torch = [ torch = [
{ index = "pytorch-cpu", extra = "cpu" }, { index = "pytorch-cpu", extra = "cpu" },
{ index = "pytorch-cu124", extra = "cu124" }, { index = "pytorch-cu128", extra = "cu128" },
] ]
torchvision = [ torchvision = [
{ index = "pytorch-cpu", extra = "cpu" }, { index = "pytorch-cpu", extra = "cpu" },
{ index = "pytorch-cu124", extra = "cu124" }, { index = "pytorch-cu128", extra = "cu128" },
] ]
[[tool.uv.index]] [[tool.uv.index]]
@ -367,8 +367,8 @@ url = "https://download.pytorch.org/whl/cpu"
explicit = true explicit = true
[[tool.uv.index]] [[tool.uv.index]]
name = "pytorch-cu124" name = "pytorch-cu128"
url = "https://download.pytorch.org/whl/cu124" url = "https://download.pytorch.org/whl/cu128"
explicit = true explicit = true
``` ```