File size: 234 Bytes
c2d8a57 | 1 2 3 4 5 6 7 8 9 10 11 12 | FROM nvcr.io/nvidia/pytorch:24.01-py3
WORKDIR /app
RUN pip install --no-cache-dir datasets huggingface_hub
COPY src/ /app/src/
COPY experiments/ /app/experiments/
COPY train_gpu.py /app/train_gpu.py
CMD ["python", "train_gpu.py"]
|