forked from MakazhanAlpamys/Soup
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathDockerfile
More file actions
33 lines (27 loc) · 1023 Bytes
/
Copy pathDockerfile
File metadata and controls
33 lines (27 loc) · 1023 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# Use NVIDIA CUDA 12.1 base image
FROM nvidia/cuda:12.1.0-runtime-ubuntu22.04
# Set environment variables
ENV DEBIAN_FRONTEND=noninteractive \
PYTHONUNBUFFERED=1 \
PYTHONIOENCODING=UTF-8
# Install Python 3.10 (Ubuntu 22.04 native — Soup supports 3.9+).
# Avoids deadsnakes PPA which has had repeated connection timeouts from GHA runners.
RUN apt-get update && apt-get install -y --no-install-recommends \
python3 \
python3-dev \
python3-venv \
python3-pip \
curl \
git \
&& rm -rf /var/lib/apt/lists/*
# Alias `python` -> `python3`
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3 1
# Set working directory
WORKDIR /workspace
# Install Soup from PyPI (always the latest published release, not local source).
# v0.71.0 split the training stack into the [train] extra — include it so the
# GPU image can still fine-tune.
RUN pip install --no-cache-dir "soup-cli[train,serve,data,eval]"
# Default entrypoint and command
ENTRYPOINT ["soup"]
CMD ["--help"]