| FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime | |
| WORKDIR /app | |
| # Install system dependencies | |
| RUN apt-get update && apt-get install -y \ | |
| git \ | |
| && rm -rf /var/lib/apt/lists/* | |
| # Install Python dependencies | |
| COPY pyproject.toml . | |
| RUN pip install poetry && \ | |
| poetry config virtualenvs.create false && \ | |
| poetry install --no-dev | |
| # Copy project files | |
| COPY src/ src/ | |
| COPY README.md . | |
| # Set environment variables | |
| ENV PYTHONPATH=/app | |
| # Default command | |
| CMD ["python", "src/train.py"] | |