in cdk-project/lib/image-stack.ts [29:73]
constructor(scope: cdk.App, prefix: string, props: ImageStackProps) {
super(scope, common.stackId(prefix, "ImageStack", props), props);
this.pullRequestBuildRole = iam.Role.fromRoleArn(
this,
"PullRequestBuildRole",
props.pullRequestBuildRole.roleArn,
);
this.codeBuildImage = this.createCodeBuildImage("sagemaker-codebuild");
this.basePythonImage = this.createProcessingImageFrom1P(
"base-python",
"sagemaker-base-python-environment",
"1.0",
);
this.dataScienceImage = this.createProcessingImageFrom1P(
"data-science",
"sagemaker-data-science-environment",
"1.0",
);
this.sparkImage = this.createProcessingImageFrom1P("spark", "sagemaker-sparkmagic", "1.0");
this.rImage = this.createProcessingImageFrom1P("r-image", "r-image", "1.0");
this.mxnetImage = this.createProcessingImageFromDlc(
"mxnet",
"mxnet-training",
"1.8.0-cpu-py37-ubuntu16.04",
);
this.pytorchImage = this.createProcessingImageFromDlc(
"pytorch",
"pytorch-training",
"1.7.1-cpu-py36-ubuntu18.04",
);
this.tensorflow1Image = this.createProcessingImageFromDlc(
"tensorflow-1",
"tensorflow-training",
"1.15.4-cpu-py37-ubuntu18.04",
);
this.tensorflow2Image = this.createProcessingImageFromDlc(
"tensorflow-2",
"tensorflow-training",
"2.4.1-cpu-py37-ubuntu18.04",
);
}