diff --git a/Dockerfile b/Dockerfile index f91c8bc..187811f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,8 +1,8 @@ # ── Platform-conditional base ───────────────────────────────────────────── -# amd64: NVIDIA CUDA 12.6 (GPU acceleration when available, CPU fallback) +# amd64: NVIDIA CUDA 13.3 (GPU acceleration when available, CPU fallback) # arm64: Plain Ubuntu (CPU-only, no CUDA on ARM) -FROM --platform=$BUILDPLATFORM nvidia/cuda:12.6.3-runtime-ubuntu22.04 AS base-amd64 +FROM --platform=$BUILDPLATFORM nvidia/cuda:13.3.0-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 29cc010..1ddfafb 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -18,7 +18,7 @@ dependencies = [ "insightface>=0.7.3", "nvidia-cudnn-cu12>=9.0.0", "numpy>=2.2.6", - "onnxruntime-gpu>=1.23.2", + "onnxruntime-gpu==1.20.1", "opencv-python-headless>=4.12.0.88", "pillow>=12.1.0", "python-dotenv>=1.2.1", @@ -37,14 +37,14 @@ Repository = "https://github.com/sudolulo/if_curator_headless" [tool.uv.sources] torch = [ - { index = "pytorch-cu126", marker = "sys_platform == 'linux' and platform_machine == 'x86_64'" }, + { index = "pytorch-cu132", 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-cu126" -url = "https://download.pytorch.org/whl/cu126" +name = "pytorch-cu132" +url = "https://download.pytorch.org/whl/cu132" explicit = true [[tool.uv.index]]