Created
June 11, 2022 05:51
-
-
Save nsakki55/d615eda7343d433c88b6fcdec8fd5fe5 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 python:3.6-slim-buster | |
RUN apt-get update \ | |
&& apt-get install -y --no-install-recommends gcc g++ | |
RUN pip3 install pandas numpy joblib scikit-learn cython | |
RUN pip3 install fastFM | |
RUN pip3 install sagemaker-training | |
ADD custom_toolkit_container_training_script.py /opt/ml/code/custom_toolkit_container_training_script.py | |
# Defines entrypoint script | |
ENV SAGEMAKER_PROGRAM custom_toolkit_container_training_script.py |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment