init

pyproject.toml

@@ -0,0 +1,63 @@
+[project]
+name = "template_project_alpha"
+version = "0.1.0"
+description = "a basic template project that has the ability to complete most of ML task with PyTorch and MMCV"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+  "anyio>=4.9.0",
+  "awkward>=2.8.5",
+  "beartype>=0.21.0",
+  "click>=8.2.1",
+  "jaxtyping>=0.3.2",
+  "loguru>=0.7.3",
+  "mmcv",
+  "mmdet>=3.3.0",
+  "mmpose>=1.3.2",
+  "nats-py>=2.10.0",
+  "numba>=0.61.2",
+  "nvidia-nvimgcodec-cu12>=0.5.0.13",
+  "opencv-python>=4.12.0.88",
+  "orjson>=3.10.18",
+  "pyarrow>=20.0.0",
+  "pydantic>=2.11.7",
+  "pytorch3d",
+  "redis>=6.2.0",
+  "result>=0.17.0",
+  "tomli>=2.2.1",
+  "tomli-w>=1.2.0",
+  "torch>=2.7.0",
+  "torchvision>=0.22.0",
+  "ultralytics>=8.3.166",
+  "xtcocotools",
+]
+
+[dependency-groups]
+dev = ["jupyterlab>=4.4.4", "pip", "psutil>=7.0.0", "setuptools"]
+
+[tool.uv]
+no-build-isolation-package = ["chumpy", "xtcocotools"]
+
+[tool.uv.sources]
+torch = [
+  { index = "pytorch-cpu", marker = "sys_platform != 'linux'" },
+  { index = "pytorch-cu128", marker = "sys_platform == 'linux'" },
+]
+torchvision = [
+  { index = "pytorch-cpu", marker = "sys_platform != 'linux'" },
+  { index = "pytorch-cu128", marker = "sys_platform == 'linux'" },
+]
+# built with cu128
+mmcv = { path = "misc/mmcv-2.2.0-cp312-cp312-linux_x86_64.whl" }
+xtcocotools = { path = "misc/xtcocotools-1.14.3-cp312-cp312-linux_x86_64.whl" }
+pytorch3d = { path = "misc/pytorch3d-0.7.8-cp312-cp312-linux_x86_64.whl" }
+
+[[tool.uv.index]]
+name = "pytorch-cpu"
+url = "https://download.pytorch.org/whl/cpu"
+explicit = true
+
+[[tool.uv.index]]
+name = "pytorch-cu128"
+url = "https://download.pytorch.org/whl/cu128"
+explicit = true