Docker: Switch image and copy config

Automatically create a config.yml on build. Also use the cuda runtime
image which is much lighter than the previous cuda devel image.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri 2024-04-15 18:01:56 -04:00
parent a2a2e4b866
commit 209f0370b4
2 changed files with 8 additions and 1 deletions

View file

@ -1 +1,4 @@
models/
loras/
.ruff_cache/
**/__pycache__/

View file

@ -1,5 +1,5 @@
# Use an official CUDA runtime with Ubuntu as a parent image
FROM nvidia/cuda:12.2.0-devel-ubuntu22.04
FROM nvidia/cuda:12.4.1-runtime-ubuntu22.04
# Set the working directory in the container
WORKDIR /app
@ -11,6 +11,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
ca-certificates \
python3.11 \
python3-pip \
git \
&& rm -rf /var/lib/apt/lists/*
# Upgrade pip
@ -25,6 +26,9 @@ RUN pip3 install --no-cache-dir .[cu121]
# Copy the current directory contents into the container
COPY . .
# Create a config.yml
COPY config_sample.yml config.yml
# Make port 5000 available to the world outside this container
EXPOSE 5000