| 1 | import * as ecr from '@aws-cdk/aws-ecr';
|
| 2 | import { BuildSpec } from './build-spec';
|
| 3 | import { BuildEnvironment, BuildImageBindOptions, BuildImageConfig, ComputeType, IBindableBuildImage, IBuildImage, ImagePullPrincipalType, IProject } from './project';
|
| 4 | import { Construct } from '@aws-cdk/core';
|
| 5 | /**
|
| 6 | * A CodeBuild GPU image running Linux.
|
| 7 | *
|
| 8 | * This class has public constants that represent the most popular GPU images from AWS Deep Learning Containers.
|
| 9 | *
|
| 10 | * @see https://aws.amazon.com/releasenotes/available-deep-learning-containers-images
|
| 11 | */
|
| 12 | export declare class LinuxGpuBuildImage implements IBindableBuildImage {
|
| 13 | private readonly repositoryName;
|
| 14 | private readonly account;
|
| 15 | /** Tensorflow 1.14.0 GPU image from AWS Deep Learning Containers. */
|
| 16 | static readonly DLC_TENSORFLOW_1_14_0: IBuildImage;
|
| 17 | /** Tensorflow 1.15.0 GPU image from AWS Deep Learning Containers. */
|
| 18 | static readonly DLC_TENSORFLOW_1_15_0: IBuildImage;
|
| 19 | /** Tensorflow 1.15.2 GPU training image from AWS Deep Learning Containers. */
|
| 20 | static readonly DLC_TENSORFLOW_1_15_2_TRAINING: IBuildImage;
|
| 21 | /** Tensorflow 1.15.2 GPU inference image from AWS Deep Learning Containers. */
|
| 22 | static readonly DLC_TENSORFLOW_1_15_2_INFERENCE: IBuildImage;
|
| 23 | /** Tensorflow 2.0.0 GPU image from AWS Deep Learning Containers. */
|
| 24 | static readonly DLC_TENSORFLOW_2_0_0: IBuildImage;
|
| 25 | /** Tensorflow 2.0.1 GPU image from AWS Deep Learning Containers. */
|
| 26 | static readonly DLC_TENSORFLOW_2_0_1: IBuildImage;
|
| 27 | /** Tensorflow 2.1.0 GPU training image from AWS Deep Learning Containers. */
|
| 28 | static readonly DLC_TENSORFLOW_2_1_0_TRAINING: IBuildImage;
|
| 29 | /** Tensorflow 2.1.0 GPU inference image from AWS Deep Learning Containers. */
|
| 30 | static readonly DLC_TENSORFLOW_2_1_0_INFERENCE: IBuildImage;
|
| 31 | /** Tensorflow 2.2.0 GPU training image from AWS Deep Learning Containers. */
|
| 32 | static readonly DLC_TENSORFLOW_2_2_0_TRAINING: IBuildImage;
|
| 33 | /** PyTorch 1.2.0 GPU image from AWS Deep Learning Containers. */
|
| 34 | static readonly DLC_PYTORCH_1_2_0: IBuildImage;
|
| 35 | /** PyTorch 1.3.1 GPU image from AWS Deep Learning Containers. */
|
| 36 | static readonly DLC_PYTORCH_1_3_1: IBuildImage;
|
| 37 | /** PyTorch 1.4.0 GPU training image from AWS Deep Learning Containers. */
|
| 38 | static readonly DLC_PYTORCH_1_4_0_TRAINING: IBuildImage;
|
| 39 | /** PyTorch 1.4.0 GPU inference image from AWS Deep Learning Containers. */
|
| 40 | static readonly DLC_PYTORCH_1_4_0_INFERENCE: IBuildImage;
|
| 41 | /** PyTorch 1.5.0 GPU training image from AWS Deep Learning Containers. */
|
| 42 | static readonly DLC_PYTORCH_1_5_0_TRAINING: IBuildImage;
|
| 43 | /** PyTorch 1.5.0 GPU inference image from AWS Deep Learning Containers. */
|
| 44 | static readonly DLC_PYTORCH_1_5_0_INFERENCE: IBuildImage;
|
| 45 | /** MXNet 1.4.1 GPU image from AWS Deep Learning Containers. */
|
| 46 | static readonly DLC_MXNET_1_4_1: IBuildImage;
|
| 47 | /** MXNet 1.6.0 GPU image from AWS Deep Learning Containers. */
|
| 48 | static readonly DLC_MXNET_1_6_0: IBuildImage;
|
| 49 | /**
|
| 50 | * Returns a Linux GPU build image from AWS Deep Learning Containers.
|
| 51 | *
|
| 52 | * @param repositoryName the name of the repository,
|
| 53 | * for example "pytorch-inference"
|
| 54 | * @param tag the tag of the image, for example "1.5.0-gpu-py36-cu101-ubuntu16.04"
|
| 55 | * @param account the AWS account ID where the DLC repository for this region is hosted in.
|
| 56 | * In many cases, the CDK can infer that for you, but for some newer region our information
|
| 57 | * might be out of date; in that case, you can specify the region explicitly using this optional parameter
|
| 58 | * @see https://aws.amazon.com/releasenotes/available-deep-learning-containers-images
|
| 59 | */
|
| 60 | static awsDeepLearningContainersImage(repositoryName: string, tag: string, account?: string): IBuildImage;
|
| 61 | /**
|
| 62 | * Returns a GPU image running Linux from an ECR repository.
|
| 63 | *
|
| 64 | * NOTE: if the repository is external (i.e. imported), then we won't be able to add
|
| 65 | * a resource policy statement for it so CodeBuild can pull the image.
|
| 66 | *
|
| 67 | * @see https://docs.aws.amazon.com/codebuild/latest/userguide/sample-ecr.html
|
| 68 | *
|
| 69 | * @param repository The ECR repository
|
| 70 | * @param tag Image tag (default "latest")
|
| 71 | */
|
| 72 | static fromEcrRepository(repository: ecr.IRepository, tag?: string): IBuildImage;
|
| 73 | readonly type = "LINUX_GPU_CONTAINER";
|
| 74 | readonly defaultComputeType = ComputeType.LARGE;
|
| 75 | readonly imagePullPrincipalType?: ImagePullPrincipalType;
|
| 76 | readonly imageId: string;
|
| 77 | private _imageAccount?;
|
| 78 | private constructor();
|
| 79 | bind(scope: Construct, project: IProject, _options: BuildImageBindOptions): BuildImageConfig;
|
| 80 | validate(buildEnvironment: BuildEnvironment): string[];
|
| 81 | runScriptBuildspec(entrypoint: string): BuildSpec;
|
| 82 | }
|