dasemu/Transcriptarr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update Dockerfile

Browse Source
This commit is contained in:
McCloudS
2025-02-03 22:59:51 -07:00
committed by GitHub
parent f835edb1f5
commit 976532a0b7
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
Dockerfile
View File

@@ -1,5 +1,5 @@
# Stage 1: Builder
FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04 as builder
FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04 AS builder
WORKDIR /subgen
@@ -13,7 +13,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
# Copy requirements and install Python dependencies
COPY requirements.txt .
RUN pip3 install --no-cache-dir -r requirements.txt
RUN pip install --no-cache-dir --prefix=/install -r requirements.txt --extra-index-url https://download.pytorch.org/wheel/cpu
# Copy application code
COPY . .
@@ -24,11 +24,10 @@ FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04
WORKDIR /subgen
# Copy necessary files from the builder stage
COPY --from=builder /subgen/requirements.txt .
COPY --from=builder /subgen/launcher.py .
COPY --from=builder /subgen/subgen.py .
COPY --from=builder /subgen/language_code.py .
COPY --from=builder /user/local /usr/local
COPY --from=builder /install /usr/local
# Install runtime dependencies
RUN apt-get update && apt-get install -y --no-install-recommends \