build: pin torch to cu130 via uv
Pin the detection stack to torch 2.11.0 and torchvision 0.26.0 and route both packages through the explicit PyTorch cu130 index for Linux and Windows installs. Move the remaining uv project configuration into pyproject.toml, set PyPI as the default index, and remove the redundant repo-local uv.toml so uv no longer emits split-configuration warnings. Refresh the lockfile against the new index configuration so the project resolves from pypi.org for general packages and download.pytorch.org/cu130 for the CUDA-specific torch artifacts.
This commit is contained in:
+20
-2
@@ -36,8 +36,8 @@ detection = [
|
||||
"nats-py>=2.11.0",
|
||||
"pydantic>=2.11.7",
|
||||
"pydantic-settings>=2.0.0",
|
||||
"torch>=2.7.0",
|
||||
"torchvision>=0.22.0",
|
||||
"torch==2.11.0",
|
||||
"torchvision==0.26.0",
|
||||
"ultralytics>=8.3.166",
|
||||
"xtcocotools",
|
||||
]
|
||||
@@ -48,12 +48,30 @@ pose-tracking-exp = "pose_tracking_exp.cli:main"
|
||||
[tool.hatch.build.targets.wheel]
|
||||
packages = ["src/pose_tracking_exp"]
|
||||
|
||||
[tool.uv]
|
||||
no-build-isolation-package = ["chumpy", "xtcocotools"]
|
||||
|
||||
[tool.uv.sources]
|
||||
rapid-pose-triangulation = { path = "../RapidPoseTriangulation", editable = true }
|
||||
cvmmap-client = { path = "../cvmmap-python-client", editable = true }
|
||||
mmcv = { path = "vendor/wheels/mmcv-2.2.0-cp312-cp312-linux_x86_64.whl" }
|
||||
torch = [
|
||||
{ index = "pytorch-cu130", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
|
||||
]
|
||||
torchvision = [
|
||||
{ index = "pytorch-cu130", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
|
||||
]
|
||||
xtcocotools = { path = "vendor/wheels/xtcocotools-1.14.3-cp312-cp312-linux_x86_64.whl" }
|
||||
|
||||
[[tool.uv.index]]
|
||||
url = "https://pypi.org/simple"
|
||||
default = true
|
||||
|
||||
[[tool.uv.index]]
|
||||
name = "pytorch-cu130"
|
||||
url = "https://download.pytorch.org/whl/cu130"
|
||||
explicit = true
|
||||
|
||||
[tool.pytest.ini_options]
|
||||
testpaths = ["tests"]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user