AWS SageMaker - Upload our own docker image on Amazon SageMaker

Question

I am new to AWS SageMaker and i am using this technology for building and training the machine learning models. I have now developed a docker image which contains our custom code for tensorflow. I would like to upload this custom docker image to AWS SageMaker and make use of it.

I have searched various links but could not find proper information on how to upload our own custom docker image.

Can you please suggest me the process of uploading our own docker image to AWS SageMaker?

Answer

Hi,
You can find detailed instructions in the [SageMaker Immersion Day](https://sagemaker-immersionday.workshop.aws/en/). In Lab3 there is a sample called "Option 2: Bring your own Container". And the corresponding notebook is [here](https://github.com/aws-samples/amazon-sagemaker-immersion-day/blob/master/bring-custom-container.ipynb). The content of the sample Docker container is inside [this](https://github.com/aws-samples/amazon-sagemaker-immersion-day/blob/master/scikit_bring_your_own.zip) zip file and showcases the decision tree algorithm from scikit-learn package.
Hope this helps.

AWS SageMaker - Upload our own docker image on Amazon SageMaker

Contenido relevante