2026-04-17 10:58:39 +08:00
|
|
|
|
FROM ghcr.io/astral-sh/uv:latest AS uv_bin
|
2026-04-17 11:28:52 +08:00
|
|
|
|
FROM nvidia/cuda:12.4.1-base-ubuntu22.04
|
2026-04-17 10:58:39 +08:00
|
|
|
|
|
|
|
|
|
|
ENV UV_LINK_MODE=copy \
|
|
|
|
|
|
UV_COMPILE_BYTECODE=1 \
|
|
|
|
|
|
PYTHONUNBUFFERED=1 \
|
|
|
|
|
|
UV_PROJECT_ENVIRONMENT=/app/.venv \
|
|
|
|
|
|
TORCH_CUDA_ARCH_LIST="8.6" \
|
|
|
|
|
|
CUDA_VISIBLE_DEVICES=0 \
|
|
|
|
|
|
UV_TORCH_BACKEND=cu128
|
|
|
|
|
|
|
|
|
|
|
|
COPY --from=uv_bin /uv /uvx /bin/
|
|
|
|
|
|
|
|
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
|
|
|
|
git \
|
|
|
|
|
|
&& apt-get clean && rm -rf /var/lib/apt/lists/*
|
|
|
|
|
|
|
|
|
|
|
|
WORKDIR /app
|
|
|
|
|
|
|
|
|
|
|
|
# 3. 安装依赖 (不加 --system，让 uv 创建受管的虚拟环境)
|
|
|
|
|
|
# 这里会根据 pyproject.toml 自动下载并安装 Python 3.11
|
|
|
|
|
|
COPY pyproject.toml uv.lock ./
|
|
|
|
|
|
RUN uv sync --frozen --no-dev --no-install-project --python 3.9
|
|
|
|
|
|
# 4. 拷贝项目文件并安装项目本身
|
|
|
|
|
|
COPY . .
|
|
|
|
|
|
RUN uv sync --frozen --no-dev --python 3.9
|
|
|
|
|
|
|
|
|
|
|
|
# 5. 【最关键】将虚拟环境的 bin 目录提到最前面
|
|
|
|
|
|
# 注意：uv sync 创建的 python 就在这个目录下
|
|
|
|
|
|
ENV PATH="/app/.venv/bin:$PATH"
|
|
|
|
|
|
|
|
|
|
|
|
EXPOSE 8000
|
|
|
|
|
|
|
|
|
|
|
|
# 验证路径并运行
|
|
|
|
|
|
# 此时运行 python 实际上是运行 /app/.venv/bin/python
|
|
|
|
|
|
CMD ["sleep", "infinity"]
|
