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

added GPU environment support

Open johnbensnyder opened this issue 3 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