Skip to content

Instantly share code, notes, and snippets.

@nsakki55

nsakki55/Dockerfile

Created June 11, 2022 05:50
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save nsakki55/f7d8917f9dc5fdf54ae29e71fad71f86 to your computer and use it in GitHub Desktop.
Save nsakki55/f7d8917f9dc5fdf54ae29e71fad71f86 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Dockerfile
FROM 354813040037.dkr.ecr.ap-northeast-1.amazonaws.com/sagemaker-scikit-learn:0.23-1-cpu-py3
RUN pip3 install --upgrade pip
RUN pip3 install optuna==2.10.0
ADD extend_sagemaker_container/extend_sagemaker_container_trainig_script.py /opt/ml/code/extend_sagemaker_container_trainig_script.py
# Defines train.py as script entrypoint
ENV SAGEMAKER_PROGRAM extend_sagemaker_container_trainig_script.py
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment