flan/winnow
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: revert to CUDA 12.6 base for onnxruntime-gpu compatibility

Browse Source
This commit is contained in:
flan
2026-06-11 09:59:53 -04:00
parent 86c85103ae
commit b0ee06ee51
2 changed files with 6 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Dockerfile
+2 -2
View File
@@ -1,8 +1,8 @@
# ── Platform-conditional base ───────────────────────────────────────────── # ── 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) # 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 FROM --platform=$BUILDPLATFORM ubuntu:22.04 AS base-arm64
# ── Build stage ─────────────────────────────────────────────────────────── # ── Build stage ───────────────────────────────────────────────────────────
pyproject.toml
+4 -26
View File
@@ -37,14 +37,14 @@ Repository = "https://github.com/sudolulo/if_curator_headless"
[tool.uv.sources] [tool.uv.sources]
torch = [ 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' and platform_machine == 'aarch64'" },
{ index = "pytorch-cpu", marker = "sys_platform != 'linux'" }, { index = "pytorch-cpu", marker = "sys_platform != 'linux'" },
] ]
[[tool.uv.index]] [[tool.uv.index]]
name = "pytorch-cu132" name = "pytorch-cu126"
url = "https://download.pytorch.org/whl/cu132" url = "https://download.pytorch.org/whl/cu126"
explicit = true explicit = true
[[tool.uv.index]] [[tool.uv.index]]
@@ -68,27 +68,5 @@ target-version = "py312"
[tool.ruff.lint] [tool.ruff.lint]
select = ["E", "F", "I"] 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"