Skip to content

Commit ceb99fb

Browse files
committed
chore: update pytorch index alias name
1 parent df882af commit ceb99fb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -302,7 +302,7 @@ torch = [
302302
# { index = "default", marker = "sys_platform == 'darwin'" }, # macOS
303303
{ index = "pytorch-arm", marker = "sys_platform != 'darwin' and platform_machine == 'arm64'" },
304304
# Linux and Windows
305-
{ index = "pytorch-cu126", marker = "sys_platform != 'darwin' and platform_machine == 'x86_64' and extra == 'using-cuda' and extra != 'using-rocm'" },
305+
{ index = "pytorch-cuda", marker = "sys_platform != 'darwin' and platform_machine == 'x86_64' and extra == 'using-cuda' and extra != 'using-rocm'" },
306306
{ index = "pytorch-rocm", marker = "sys_platform != 'darwin' and platform_machine == 'x86_64' and extra != 'using-cuda' and extra == 'using-rocm'" },
307307
{ index = "pytorch-cpu", marker = "sys_platform != 'darwin' and platform_machine == 'x86_64' and extra != 'using-cuda' and extra != 'using-rocm'" },
308308
]
@@ -313,7 +313,7 @@ url = "https://download.pytorch.org/whl/rocm6.2"
313313
explicit = true
314314

315315
[[tool.uv.index]]
316-
name = "pytorch-cu126"
316+
name = "pytorch-cuda"
317317
url = "https://download.pytorch.org/whl/cu126"
318318
explicit = true
319319

0 commit comments

Comments
 (0)