7d11dfa688
Implementation of a mock scheduler for jobs/tasks that will evolve into a job manager for running training jobs over vast.ai instances.
15 lines
410 B
Docker
15 lines
410 B
Docker
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"]
|