Up
This commit is contained in:
+3
-3
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user