sagemaker-studio-image-build-cli icon indicating copy to clipboard operation
sagemaker-studio-image-build-cli copied to clipboard
verified publisher icon aws-samples

added GPU environment support

Open johnbensnyder opened this issue 4 years ago • 1 comments

Issue #, if available:

Issue https://github.com/aws-samples/sagemaker-studio-image-build-cli/issues/10

Description of changes:

Adds --environment option to set LINUX_GPU_CONTAINER when build involves compiling Cuda operations.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

johnbensnyder avatar Apr 05 '21 01:04 johnbensnyder

Sorry for the long delay. I've updated the changes so --compute-type now has to be manually changed by the user for GPU instances.

johnbensnyder avatar Aug 29 '21 23:08 johnbensnyder