| FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime | |
| WORKDIR /app | |
| COPY requirements.txt requirements-train.txt ./ | |
| RUN pip install --no-cache-dir -r requirements.txt -r requirements-train.txt | |
| COPY . . | |
| CMD ["python", "-m", "training.generate_data", "--episodes", "2000"] | |