File size: 271 Bytes
698f4d8 | 1 2 3 4 5 6 7 8 9 10 11 | 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"]
|