Skip to content

Instantly share code, notes, and snippets.

@nsakki55

nsakki55/Dockerfile

Created June 11, 2022 05:51
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/d615eda7343d433c88b6fcdec8fd5fe5 to your computer and use it in GitHub Desktop.
Save nsakki55/d615eda7343d433c88b6fcdec8fd5fe5 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Dockerfile
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