Add cu126 sources

This commit is contained in:
mbsantiago 2026-03-25 13:22:24 +00:00
parent 6bad018045
commit 85667ca676

View File

@ -60,6 +60,22 @@ dev = [
"ruff>=0.7.3", "ruff>=0.7.3",
] ]
[[tool.uv.index]]
name = "pytorch-cu126"
url = "https://download.pytorch.org/whl/cu126"
explicit = true
[tool.uv.sources]
torch = [
{ index = "pytorch-cu126", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
]
torchvision = [
{ index = "pytorch-cu126", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
]
torchaudio = [
{ index = "pytorch-cu126", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
]
[tool.ruff] [tool.ruff]
line-length = 79 line-length = 79
target-version = "py39" target-version = "py39"