UNPKG

aws-cdk-neuronx-patterns

Version:
42 lines (41 loc) 1.57 kB
import { Size } from "aws-cdk-lib"; export interface INeuronxImage { readonly imageName: string; readonly imageTag: string; readonly sdkVersion: string; } declare abstract class NeuronxImage { static readonly size: Size; /** Neuron SDK 2.23.0 */ static readonly SDK_2_23_0: INeuronxImage; /** Neuron SDK 2.22.0 */ static readonly SDK_2_22_0: INeuronxImage; /** Neuron SDK 2.21.1 */ static readonly SDK_2_21_1: INeuronxImage; /** Neuron SDK 2.21.0 */ static readonly SDK_2_21_0: INeuronxImage; /** Neuron SDK 2.20.2 */ static readonly SDK_2_20_2: INeuronxImage; /** Neuron SDK 2.20.1 */ static readonly SDK_2_20_1: INeuronxImage; /** Neuron SDK 2.20.0 */ static readonly SDK_2_20_0: INeuronxImage; /** Neuron SDK 2.19.1 */ static readonly SDK_2_19_1: INeuronxImage; /** Neuron SDK 2.19.0 */ static readonly SDK_2_19_0: INeuronxImage; /** Neuron SDK 2.18.2 */ static readonly SDK_2_18_2: INeuronxImage; /** Neuron SDK 2.18.1 */ static readonly SDK_2_18_1: INeuronxImage; /** Neuron SDK 2.18.0 */ static readonly SDK_2_18_0: INeuronxImage; /** Latest Neuron SDK */ static readonly LATEST: INeuronxImage; static fromSdkVersion(neuronxVersion: string, pythonVersion: string, pytorchVersion: string, ubuntuVersion: string): INeuronxImage; protected static readonly imageName: string; } export declare class PytorchTrainingNeuronxImage extends NeuronxImage { static readonly imageName = "public.ecr.aws/neuron/pytorch-training-neuronx"; } export {};