diff --git a/Dockerfile b/Dockerfile index 187811f..27d9582 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,8 +1,8 @@ # ── Platform-conditional base ───────────────────────────────────────────── -# amd64: NVIDIA CUDA 13.3 (GPU acceleration when available, CPU fallback) +# amd64: NVIDIA CUDA 12.6 (GPU acceleration when available, CPU fallback) # arm64: Plain Ubuntu (CPU-only, no CUDA on ARM) -FROM --platform=$BUILDPLATFORM nvidia/cuda:13.3.0-cudnn-runtime-ubuntu22.04 AS base-amd64 +FROM --platform=$BUILDPLATFORM nvidia/cuda:12.6.3-cudnn-runtime-ubuntu22.04 AS base-amd64 FROM --platform=$BUILDPLATFORM ubuntu:22.04 AS base-arm64 # ── Build stage ─────────────────────────────────────────────────────────── diff --git a/pyproject.toml b/pyproject.toml index 1ddfafb..48fcebf 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -37,14 +37,14 @@ Repository = "https://github.com/sudolulo/if_curator_headless" [tool.uv.sources] torch = [ - { index = "pytorch-cu132", marker = "sys_platform == 'linux' and platform_machine == 'x86_64'" }, + { index = "pytorch-cu126", marker = "sys_platform == 'linux' and platform_machine == 'x86_64'" }, { index = "pytorch-cpu", marker = "sys_platform == 'linux' and platform_machine == 'aarch64'" }, { index = "pytorch-cpu", marker = "sys_platform != 'linux'" }, ] [[tool.uv.index]] -name = "pytorch-cu132" -url = "https://download.pytorch.org/whl/cu132" +name = "pytorch-cu126" +url = "https://download.pytorch.org/whl/cu126" explicit = true [[tool.uv.index]] @@ -68,27 +68,5 @@ target-version = "py312" [tool.ruff.lint] select = ["E", "F", "I"] -[tool.deptry] -pep621_dev_dependency_groups = ["dev"] - -[tool.deptry.package_module_name_map] -pillow = "PIL" -opencv-python-headless = "cv2" -python-dotenv = "dotenv" -insightface = "insightface" -numpy = "numpy" -nvidia-cudnn-cu12 = "nvidia.cudnn" -onnxruntime-gpu = "onnxruntime" -requests = "requests" -rich = "rich" -torch = "torch" -transformers = "transformers" -ultralytics = "ultralytics" - -[tool.deptry.per_rule_ignores] -DEP002 = ["onnxruntime-gpu", "nvidia-cudnn-cu12"] - -[build-system] -requires = ["hatchling"] -build-backend = "hatchling.build" +[