apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-06-08 20:25:19 +02:00
Code Issues Projects Releases Packages Wiki Activity

feat(pyproject): add support for CUDA 12.6 in optional dependencies

Browse source 
- Introduced a new optional dependency group for CUDA 12.6 in pyproject.toml.
- Updated conflict resolution to include CUDA 12.6 alongside existing configurations.
- Added source configurations for PyTorch and torchvision for CUDA 12.6.
This commit is contained in:
Anish Sarkar 2026-06-05 22:25:32 +05:30
parent 6972356c86
commit f0ef3c02ac
1 changed files with 9 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

10
surfsense_backend/pyproject.toml
View file

@ -94,6 +94,7 @@ dependencies = [
[project.optional-dependencies]
cpu = ["torch==2.11.0", "torchvision==0.26.0"]
cu126 = ["torch==2.11.0", "torchvision==0.26.0"]
cu128 = ["torch==2.11.0", "torchvision==0.26.0"]
[dependency-groups]
@ -106,15 +107,17 @@ dev = [
]
[tool.uv]
conflicts = [[{ extra = "cpu" }, { extra = "cu128" }]]
conflicts = [[{ extra = "cpu" }, { extra = "cu126" }, { extra = "cu128" }]]
[tool.uv.sources]
torch = [
{ index = "pytorch-cpu", extra = "cpu", marker = "sys_platform == 'linux'" },
{ index = "pytorch-cu126", extra = "cu126", marker = "sys_platform == 'linux'" },
{ index = "pytorch-cu128", extra = "cu128", marker = "sys_platform == 'linux'" },
]
torchvision = [
{ index = "pytorch-cpu", extra = "cpu", marker = "sys_platform == 'linux'" },
{ index = "pytorch-cu126", extra = "cu126", marker = "sys_platform == 'linux'" },
{ index = "pytorch-cu128", extra = "cu128", marker = "sys_platform == 'linux'" },
]
@ -123,6 +126,11 @@ name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
explicit = true
[[tool.uv.index]]
name = "pytorch-cu126"
url = "https://download.pytorch.org/whl/cu126"
explicit = true
[[tool.uv.index]]
name = "pytorch-cu128"
url = "https://download.pytorch.org/whl/cu128"