anugrahteesdollar commited on
Commit
7e81b32
·
verified ·
1 Parent(s): e681925

space: add root Dockerfile for trainer Space

Browse files
Files changed (1) hide show
  1. Dockerfile +36 -0
Dockerfile ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # DrugEnv trainer Space (Docker, single H200 GPU)
2
+ # Serves the FastAPI control panel (space.training.app:app) on port 8000,
3
+ # matched by README YAML app_port: 8000.
4
+ FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
5
+
6
+ ENV DEBIAN_FRONTEND=noninteractive \
7
+ PYTHONUNBUFFERED=1 \
8
+ PIP_NO_CACHE_DIR=1 \
9
+ HF_HOME=/home/user/.cache/huggingface \
10
+ TRANSFORMERS_CACHE=/home/user/.cache/huggingface/transformers \
11
+ PYTHONPATH=/home/user/app \
12
+ PORT=8000
13
+
14
+ RUN apt-get update && apt-get install -y --no-install-recommends \
15
+ python3.11 python3.11-venv python3.11-dev python3-pip \
16
+ git curl ca-certificates build-essential \
17
+ && rm -rf /var/lib/apt/lists/* \
18
+ && ln -sf /usr/bin/python3.11 /usr/local/bin/python \
19
+ && ln -sf /usr/bin/python3.11 /usr/local/bin/python3
20
+
21
+ RUN useradd -ms /bin/bash user
22
+ USER user
23
+ ENV PATH="/home/user/.local/bin:${PATH}"
24
+ WORKDIR /home/user/app
25
+
26
+ # Copy the entire repo first so relative -r references inside the
27
+ # trainer requirements file (-r ../../requirements-train.txt etc.)
28
+ # resolve correctly. Only after the tree is in place do we install.
29
+ COPY --chown=user:user . /home/user/app
30
+
31
+ RUN python -m pip install --upgrade pip && \
32
+ python -m pip install --user -r /home/user/app/space/training/requirements.txt
33
+
34
+ EXPOSE 8000
35
+
36
+ CMD ["python", "-m", "uvicorn", "space.training.app:app", "--host", "0.0.0.0", "--port", "8000"]