This commit is contained in:
Ryan
2026-08-10 21:00:22 -04:00
parent 89d93a5a20
commit 1540e30be5
3 changed files with 27 additions and 24 deletions
+3 -3
View File
@@ -1,6 +1,5 @@
# Base image: CUDA 11.8 runtime on Ubuntu 22.04
#FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
FROM nvidia/cuda:13.3.1-cudnn-runtime-ubuntu22.04
FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
# Avoid interactive prompts during build
ENV DEBIAN_FRONTEND=noninteractive
@@ -36,7 +35,8 @@ RUN groupadd -g 1000 appuser && \
WORKDIR /app
# Create necessary directories
RUN mkdir -p /scratch /models /data/training /data/output /logs
RUN mkdir -p /scratch /models /data/training /data/output /logs && \
chown -R appuser:appuser /app /scratch /models /data /logs
# Copy requirements first for better caching
COPY worker/requirements.txt /app/requirements.txt