File size: 449 Bytes
af6e330 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | ARG BASE_DOCKER
# Dockerfile that updates the container with new code.
# SageMaker PyTorch image
FROM ${BASE_DOCKER}
# /opt/ml and all subdirectories are utilized by SageMaker, use the /code subdirectory to store your user code.
COPY . /opt/ml/code/
# RUN pip install -e /opt/ml/code/
# Prevent sagemaker from installing requirements again.
RUN rm /opt/ml/code/setup.py
RUN rm /opt/ml/code/requirements.txt
ENV SAGEMAKER_PROGRAM open_lm/main.py
|