File size: 336 Bytes
ee77ed4 640e519 ee77ed4 766903e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
RUN apt-get update && \
apt-get install -y stockfish && \
rm -rf /var/lib/apt/lists/*
WORKDIR /app
COPY requirements_space.txt .
RUN pip install --no-cache-dir -r requirements_space.txt
COPY train_hf.py .
COPY finetune_hf.py .
COPY src/ src/
CMD ["python", "train_hf.py"]
|