Created
June 11, 2022 05:50
-
-
Save nsakki55/f7d8917f9dc5fdf54ae29e71fad71f86 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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