sagemaker-studio-image-build-cli
sagemaker-studio-image-build-cli copied to clipboard
Set build environment type as new arg
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 ?
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.