16 lines
410 B
Text
16 lines
410 B
Text
|
FROM python:3.13-slim
|
||
|
|
|
||
|
|
RUN apt-get update && apt-get install -y openssh-client && rm -rf /var/lib/apt/lists/*
|
||
|
|
|
||
|
|
WORKDIR /app
|
||
|
|
|
||
|
|
COPY pyproject.toml ./
|
||
|
|
COPY sched/ ./sched/
|
||
|
|
COPY worker/ ./worker/
|
||
|
|
COPY jobs/ ./jobs/
|
||
|
|
COPY tests/ ./tests/
|
||
|
|
|
||
|
|
RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu
|
||
|
|
|
||
|
|
CMD ["python", "-m", "sched", "run", "--script", "jobs/mnist.py", "--compose", "--trace"]
|