kapitan/wdtagger
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(uv): add cuda11 and cuda12 support

Browse Source
This commit is contained in:
Jianqi Pan 2025-02-21 15:07:59 +09:00
parent b461680bc0
commit 243d1c802a
1 changed files with 22 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
pyproject.toml
View File

@ -39,18 +39,29 @@ ignore = [
[tool.pyright]
[tool.uv]
conflicts = [[{ extra = "cpu" }, { extra = "gpu" }]]
conflicts = [
[
{ extra = "cpu" },
{ extra = "gpu" },
{ extra = "cuda11" },
{ extra = "cuda12" },
],
]
package = true
[tool.uv.sources]
torch = [
{ index = "torch-cpu", extra = "cpu" },
{ index = "torch-gpu", extra = "gpu" },
{ index = "torch-cuda11", extra = "cuda11" },
{ index = "torch-cuda12", extra = "cuda12" },
]
torchvision = [
{ index = "torch-cpu", extra = "cpu" },
{ index = "torch-gpu", extra = "gpu" },
{ index = "torch-cuda11", extra = "cuda11" },
{ index = "torch-cuda12", extra = "cuda12" },
]
[[tool.uv.index]]
@ -63,6 +74,16 @@ name = "torch-gpu"
url = "https://download.pytorch.org/whl/cu124"
explicit = true
[[tool.uv.index]]
name = "torch-cuda11"
url = "https://download.pytorch.org/whl/cu118"
explicit = true
[[tool.uv.index]]
name = "torch-cuda12"
url = "https://download.pytorch.org/whl/cu124"
explicit = true
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"