diff --git a/pyproject.toml b/pyproject.toml index 5b26bdd..20cdab9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -57,11 +57,15 @@ dev = [ "pyinstaller>=6.17.0", ] -# Use PyTorch CUDA index by default +# Add PyTorch CUDA index as additional source # CUDA builds work on both GPU and CPU systems (fallback to CPU if no GPU) [[tool.uv.index]] +name = "pytorch-cu121" url = "https://download.pytorch.org/whl/cu121" -default = true + +# Tell uv to get torch from the PyTorch CUDA index +[tool.uv.sources] +torch = { index = "pytorch-cu121" } [tool.ruff] line-length = 100