64 lines
1.6 KiB
TOML
64 lines
1.6 KiB
TOML
[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
|