aws-samples / aws-samples/sagemaker-studio-image-build-cli
Add support for GPU build environment
Open
enhancement
- Dominant language
- Python
- Stars
- 59
- Forks
- 26
- PR merge metrics
- No merged PRs in 30d
Description
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`.
Contributor guide
Assessment
This issue has not been assessed yet.