sagemaker-studio-image-build-cli icon indicating copy to clipboard operation
sagemaker-studio-image-build-cli copied to clipboard

Set build environment type as new arg

Open mattmcclean opened this issue 5 years ago • 1 comments

When building a conda environment in the Dockerfile I am getting the following error message:

CondaMemoryError: The conda process ran out of memory. Increase system memory and/or try again.

Would it be possible to have an argument to set the build environment type as described here ?

mattmcclean avatar Dec 13 '20 15:12 mattmcclean

Hi @mattmcclean , yes that's feasible as an optional CLI argument. Currently, the CLI is fixing to BUILD_GENERAL1_SMALL https://github.com/aws-samples/sagemaker-studio-image-build-cli/blob/master/sagemaker_studio_image_build/codebuild.py#L65

Feel free to file a PR, if you wish, and I can help review.

jaipreet-s avatar Dec 15 '20 04:12 jaipreet-s