sagemaker-pytorch-training-toolkit icon indicating copy to clipboard operation
sagemaker-pytorch-training-toolkit copied to clipboard

Example use case

Open akinolawilson opened this issue 4 years ago • 2 comments

would it possible to have an example use case of this repository?

Would I clone this whilst in the Sagemaker studio? Would it be possible to build an image from this repository and then push it up to the ECR service, following that mount it as an image to my Sagemaker Studio?

akinolawilson avatar Feb 25 '21 21:02 akinolawilson

Hey @akinolawilson,

Thanks for the suggestion!

would it possible to have an example use case of this repository?

As of now, this repository hosts the container side code for enabling training within the SageMaker PyTorch container with the SageMaker Python SDK.

For example, all of the PyTorch examples using the Python SDK for training utilize containers generated from this repository: https://github.com/aws/amazon-sagemaker-examples/tree/master/sagemaker-python-sdk

Would I clone this whilst in the Sagemaker studio? Would it be possible to build an image from this repository and then push it up to the ECR service, following that mount it as an image to my Sagemaker Studio?

For using this image in Studio, I would recommend following:

  • https://docs.aws.amazon.com/sagemaker/latest/dg/studio-byoi.html
  • https://github.com/aws-samples/sagemaker-studio-custom-image-samples

You would need to do the following:

  1. Build the image from this repository
  2. Push the image to ECR
  3. Create a SageMaker Image and ImageVersion mapping to the ECR Image
  4. Create a SageMaker Studio AppImageConfig
  5. Create/Update your SageMaker Domain

Similar to: https://github.com/aws-samples/sagemaker-studio-custom-image-samples/tree/main/examples/r-image.

You can also extend an existing image, if you don't wish to build the image from this repository.

https://docs.aws.amazon.com/sagemaker/latest/dg/prebuilt-containers-extend.html

Thanks!

ChoiByungWook avatar Feb 26 '21 03:02 ChoiByungWook

@ChoiByungWook thank you so much!

akinolawilson avatar Feb 26 '21 09:02 akinolawilson