From 20295d045700a9112f52dd2a4d8d9a5e69f1c0c6 Mon Sep 17 00:00:00 2001 From: mbsantiago Date: Wed, 25 Mar 2026 13:42:09 +0000 Subject: [PATCH] Add cu118 for python 3.9 --- pyproject.toml | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/pyproject.toml b/pyproject.toml index 72d221c..5954a25 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -67,14 +67,22 @@ name = "pytorch-cu126" url = "https://download.pytorch.org/whl/cu126" explicit = true +[[tool.uv.index]] +name = "pytorch-cu118" +url = "https://download.pytorch.org/whl/cu118" +explicit = true + [tool.uv.sources] torch = [ + { index = "pytorch-cu118", marker = "python_version == '3.9' and (sys_platform == 'linux' or sys_platform == 'win32')" }, { index = "pytorch-cu126", marker = "sys_platform == 'linux' or sys_platform == 'win32'" }, ] torchvision = [ + { index = "pytorch-cu118", marker = "python_version == '3.9' and (sys_platform == 'linux' or sys_platform == 'win32')" }, { index = "pytorch-cu126", marker = "sys_platform == 'linux' or sys_platform == 'win32'" }, ] torchaudio = [ + { index = "pytorch-cu118", marker = "python_version == '3.9' and (sys_platform == 'linux' or sys_platform == 'win32')" }, { index = "pytorch-cu126", marker = "sys_platform == 'linux' or sys_platform == 'win32'" }, ]