aws-samples / aws-samples/sagemaker-studio-image-build-cli

Add support for GPU build environment

Open
#10 0 comments 0 reactions 0 assignees View on GitHub
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

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.