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

Published 20 hours ago • verified publisher icon aws-samples

Add support for GPU build environment

Open johnbensnyder opened this issue 3 years ago • 0 comments

Currently build environment is hardcoded to LINUX_CONTAINER. This will cause builds that involve compiling Cuda operations to fail (for example, Docker images that include custom PyTorch or Tensorflow operations). Adding an --environment option to change build environment to LINUX_GPU_CONTAINER fixes this. This also requires changing the --compute-type argument to BUILD_GENERAL1_LARGE.

johnbensnyder avatar Apr 05 '21 01:04 johnbensnyder