aws-cdk-neuronx-patterns
Version:
> [!WARNING] > This library is experimental module.
52 lines • 9.4 kB
JavaScript
;
var _a;
Object.defineProperty(exports, "__esModule", { value: true });
exports.PytorchTrainingNeuronxImage = void 0;
const JSII_RTTI_SYMBOL_1 = Symbol.for("jsii.rtti");
const aws_cdk_lib_1 = require("aws-cdk-lib");
class NeuronxImage {
static fromSdkVersion(neuronxVersion, pythonVersion, pytorchVersion, ubuntuVersion) {
return {
sdkVersion: neuronxVersion,
imageName: this.imageName,
imageTag: `${pytorchVersion}-neuronx-py${pythonVersion.replace(".", "")}-sdk${neuronxVersion}-ubuntu${ubuntuVersion}`,
};
}
}
NeuronxImage.size = aws_cdk_lib_1.Size.gibibytes(30);
/** Neuron SDK 2.23.0 */
NeuronxImage.SDK_2_23_0 = NeuronxImage.fromSdkVersion("2.23.0", "3.10", "2.6.0", "22.04");
/** Neuron SDK 2.22.0 */
NeuronxImage.SDK_2_22_0 = NeuronxImage.fromSdkVersion("2.22.0", "3.10", "2.5.1", "22.04");
/** Neuron SDK 2.21.1 */
NeuronxImage.SDK_2_21_1 = NeuronxImage.fromSdkVersion("2.21.1", "3.10", "2.5.1", "22.04");
/** Neuron SDK 2.21.0 */
NeuronxImage.SDK_2_21_0 = NeuronxImage.fromSdkVersion("2.21.0", "3.10", "2.5.1", "22.04");
/** Neuron SDK 2.20.2 */
NeuronxImage.SDK_2_20_2 = NeuronxImage.fromSdkVersion("2.20.2", "3.10", "2.1.2", "20.04");
/** Neuron SDK 2.20.1 */
NeuronxImage.SDK_2_20_1 = NeuronxImage.fromSdkVersion("2.20.1", "3.10", "2.1.2", "20.04");
/** Neuron SDK 2.20.0 */
NeuronxImage.SDK_2_20_0 = NeuronxImage.fromSdkVersion("2.20.0", "3.10", "2.1.2", "20.04");
/** Neuron SDK 2.19.1 */
NeuronxImage.SDK_2_19_1 = NeuronxImage.fromSdkVersion("2.19.1", "3.10", "2.1.2", "20.04");
/** Neuron SDK 2.19.0 */
NeuronxImage.SDK_2_19_0 = NeuronxImage.fromSdkVersion("2.19.0", "3.10", "2.1.2", "20.04");
/** Neuron SDK 2.18.2 */
NeuronxImage.SDK_2_18_2 = NeuronxImage.fromSdkVersion("2.18.2", "3.10", "2.1.2", "20.04");
/** Neuron SDK 2.18.1 */
NeuronxImage.SDK_2_18_1 = NeuronxImage.fromSdkVersion("2.18.1", "3.10", "2.1.2", "20.04");
/** Neuron SDK 2.18.0 */
NeuronxImage.SDK_2_18_0 = NeuronxImage.fromSdkVersion("2.18.0", "3.10", "2.1.2", "20.04");
/** Latest Neuron SDK */
NeuronxImage.LATEST = Object.entries(NeuronxImage)
.filter(([k]) => k.startsWith("SDK_"))
.sort(([k1], [k2]) => (k1 < k2 ? 1 : -1))
.map(([_, v]) => v)[0];
class PytorchTrainingNeuronxImage extends NeuronxImage {
}
exports.PytorchTrainingNeuronxImage = PytorchTrainingNeuronxImage;
_a = JSII_RTTI_SYMBOL_1;
PytorchTrainingNeuronxImage[_a] = { fqn: "aws-cdk-neuronx-patterns.PytorchTrainingNeuronxImage", version: "0.0.23" };
PytorchTrainingNeuronxImage.imageName = "public.ecr.aws/neuron/pytorch-training-neuronx";
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiZGVlcC1sZWFybmluZy1jb250YWluZXJzLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vLi4vc3JjL2Jhc2UvbmV1cm9ueC9kZWVwLWxlYXJuaW5nLWNvbnRhaW5lcnMudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6Ijs7Ozs7QUFBQSw2Q0FBbUM7QUFPbkMsTUFBZSxZQUFZO0lBNEZ6QixNQUFNLENBQUMsY0FBYyxDQUNuQixjQUFzQixFQUN0QixhQUFxQixFQUNyQixjQUFzQixFQUN0QixhQUFxQjtRQUVyQixPQUFPO1lBQ0wsVUFBVSxFQUFFLGNBQWM7WUFDMUIsU0FBUyxFQUFFLElBQUksQ0FBQyxTQUFTO1lBQ3pCLFFBQVEsRUFBRSxHQUFHLGNBQWMsY0FBYyxhQUFhLENBQUMsT0FBTyxDQUFDLEdBQUcsRUFBRSxFQUFFLENBQUMsT0FBTyxjQUFjLFVBQVUsYUFBYSxFQUFFO1NBQ3RILENBQUM7SUFDSixDQUFDOztBQXRHZSxpQkFBSSxHQUFHLGtCQUFJLENBQUMsU0FBUyxDQUFDLEVBQUUsQ0FBQyxDQUFDO0FBQzFDLHdCQUF3QjtBQUNSLHVCQUFVLEdBQUcsWUFBWSxDQUFDLGNBQWMsQ0FDdEQsUUFBUSxFQUNSLE1BQU0sRUFDTixPQUFPLEVBQ1AsT0FBTyxDQUNSLENBQUM7QUFDRix3QkFBd0I7QUFDUix1QkFBVSxHQUFHLFlBQVksQ0FBQyxjQUFjLENBQ3RELFFBQVEsRUFDUixNQUFNLEVBQ04sT0FBTyxFQUNQLE9BQU8sQ0FDUixDQUFDO0FBQ0Ysd0JBQXdCO0FBQ1IsdUJBQVUsR0FBRyxZQUFZLENBQUMsY0FBYyxDQUN0RCxRQUFRLEVBQ1IsTUFBTSxFQUNOLE9BQU8sRUFDUCxPQUFPLENBQ1IsQ0FBQztBQUNGLHdCQUF3QjtBQUNSLHVCQUFVLEdBQUcsWUFBWSxDQUFDLGNBQWMsQ0FDdEQsUUFBUSxFQUNSLE1BQU0sRUFDTixPQUFPLEVBQ1AsT0FBTyxDQUNSLENBQUM7QUFDRix3QkFBd0I7QUFDUix1QkFBVSxHQUFHLFlBQVksQ0FBQyxjQUFjLENBQ3RELFFBQVEsRUFDUixNQUFNLEVBQ04sT0FBTyxFQUNQLE9BQU8sQ0FDUixDQUFDO0FBQ0Ysd0JBQXdCO0FBQ1IsdUJBQVUsR0FBRyxZQUFZLENBQUMsY0FBYyxDQUN0RCxRQUFRLEVBQ1IsTUFBTSxFQUNOLE9BQU8sRUFDUCxPQUFPLENBQ1IsQ0FBQztBQUNGLHdCQUF3QjtBQUNSLHVCQUFVLEdBQUcsWUFBWSxDQUFDLGNBQWMsQ0FDdEQsUUFBUSxFQUNSLE1BQU0sRUFDTixPQUFPLEVBQ1AsT0FBTyxDQUNSLENBQUM7QUFDRix3QkFBd0I7QUFDUix1QkFBVSxHQUFHLFlBQVksQ0FBQyxjQUFjLENBQ3RELFFBQVEsRUFDUixNQUFNLEVBQ04sT0FBTyxFQUNQLE9BQU8sQ0FDUixDQUFDO0FBQ0Ysd0JBQXdCO0FBQ1IsdUJBQVUsR0FBRyxZQUFZLENBQUMsY0FBYyxDQUN0RCxRQUFRLEVBQ1IsTUFBTSxFQUNOLE9BQU8sRUFDUCxPQUFPLENBQ1IsQ0FBQztBQUNGLHdCQUF3QjtBQUNSLHVCQUFVLEdBQUcsWUFBWSxDQUFDLGNBQWMsQ0FDdEQsUUFBUSxFQUNSLE1BQU0sRUFDTixPQUFPLEVBQ1AsT0FBTyxDQUNSLENBQUM7QUFDRix3QkFBd0I7QUFDUix1QkFBVSxHQUFHLFlBQVksQ0FBQyxjQUFjLENBQ3RELFFBQVEsRUFDUixNQUFNLEVBQ04sT0FBTyxFQUNQLE9BQU8sQ0FDUixDQUFDO0FBQ0Ysd0JBQXdCO0FBQ1IsdUJBQVUsR0FBRyxZQUFZLENBQUMsY0FBYyxDQUN0RCxRQUFRLEVBQ1IsTUFBTSxFQUNOLE9BQU8sRUFDUCxPQUFPLENBQ1IsQ0FBQztBQUNGLHdCQUF3QjtBQUNSLG1CQUFNLEdBQUcsTUFBTSxDQUFDLE9BQU8sQ0FBQyxZQUFZLENBQUM7S0FDbEQsTUFBTSxDQUFDLENBQUMsQ0FBQyxDQUFDLENBQUMsRUFBRSxFQUFFLENBQUMsQ0FBQyxDQUFDLFVBQVUsQ0FBQyxNQUFNLENBQUMsQ0FBQztLQUNyQyxJQUFJLENBQUMsQ0FBQyxDQUFDLEVBQUUsQ0FBQyxFQUFFLENBQUMsRUFBRSxDQUFDLEVBQUUsRUFBRSxDQUFDLENBQUMsRUFBRSxHQUFHLEVBQUUsQ0FBQyxDQUFDLENBQUMsQ0FBQyxDQUFDLENBQUMsQ0FBQyxDQUFDLENBQUMsQ0FBQyxDQUFDO0tBQ3hDLEdBQUcsQ0FBQyxDQUFDLENBQUMsQ0FBQyxFQUFFLENBQUMsQ0FBQyxFQUFFLEVBQUUsQ0FBQyxDQUFDLENBQUMsQ0FBQyxDQUFDLENBQWtCLENBQUM7QUFpQjVDLE1BQWEsMkJBQTRCLFNBQVEsWUFBWTs7QUFBN0Qsa0VBRUM7OztBQURpQixxQ0FBUyxHQUFHLGdEQUFnRCxDQUFDIiwic291cmNlc0NvbnRlbnQiOlsiaW1wb3J0IHsgU2l6ZSB9IGZyb20gXCJhd3MtY2RrLWxpYlwiO1xuXG5leHBvcnQgaW50ZXJmYWNlIElOZXVyb254SW1hZ2Uge1xuICByZWFkb25seSBpbWFnZU5hbWU6IHN0cmluZztcbiAgcmVhZG9ubHkgaW1hZ2VUYWc6IHN0cmluZztcbiAgcmVhZG9ubHkgc2RrVmVyc2lvbjogc3RyaW5nO1xufVxuYWJzdHJhY3QgY2xhc3MgTmV1cm9ueEltYWdlIHtcbiAgc3RhdGljIHJlYWRvbmx5IHNpemUgPSBTaXplLmdpYmlieXRlcygzMCk7XG4gIC8qKiBOZXVyb24gU0RLIDIuMjMuMCAqL1xuICBzdGF0aWMgcmVhZG9ubHkgU0RLXzJfMjNfMCA9IE5ldXJvbnhJbWFnZS5mcm9tU2RrVmVyc2lvbihcbiAgICBcIjIuMjMuMFwiLFxuICAgIFwiMy4xMFwiLFxuICAgIFwiMi42LjBcIixcbiAgICBcIjIyLjA0XCIsXG4gICk7XG4gIC8qKiBOZXVyb24gU0RLIDIuMjIuMCAqL1xuICBzdGF0aWMgcmVhZG9ubHkgU0RLXzJfMjJfMCA9IE5ldXJvbnhJbWFnZS5mcm9tU2RrVmVyc2lvbihcbiAgICBcIjIuMjIuMFwiLFxuICAgIFwiMy4xMFwiLFxuICAgIFwiMi41LjFcIixcbiAgICBcIjIyLjA0XCIsXG4gICk7XG4gIC8qKiBOZXVyb24gU0RLIDIuMjEuMSAqL1xuICBzdGF0aWMgcmVhZG9ubHkgU0RLXzJfMjFfMSA9IE5ldXJvbnhJbWFnZS5mcm9tU2RrVmVyc2lvbihcbiAgICBcIjIuMjEuMVwiLFxuICAgIFwiMy4xMFwiLFxuICAgIFwiMi41LjFcIixcbiAgICBcIjIyLjA0XCIsXG4gICk7XG4gIC8qKiBOZXVyb24gU0RLIDIuMjEuMCAqL1xuICBzdGF0aWMgcmVhZG9ubHkgU0RLXzJfMjFfMCA9IE5ldXJvbnhJbWFnZS5mcm9tU2RrVmVyc2lvbihcbiAgICBcIjIuMjEuMFwiLFxuICAgIFwiMy4xMFwiLFxuICAgIFwiMi41LjFcIixcbiAgICBcIjIyLjA0XCIsXG4gICk7XG4gIC8qKiBOZXVyb24gU0RLIDIuMjAuMiAqL1xuICBzdGF0aWMgcmVhZG9ubHkgU0RLXzJfMjBfMiA9IE5ldXJvbnhJbWFnZS5mcm9tU2RrVmVyc2lvbihcbiAgICBcIjIuMjAuMlwiLFxuICAgIFwiMy4xMFwiLFxuICAgIFwiMi4xLjJcIixcbiAgICBcIjIwLjA0XCIsXG4gICk7XG4gIC8qKiBOZXVyb24gU0RLIDIuMjAuMSAqL1xuICBzdGF0aWMgcmVhZG9ubHkgU0RLXzJfMjBfMSA9IE5ldXJvbnhJbWFnZS5mcm9tU2RrVmVyc2lvbihcbiAgICBcIjIuMjAuMVwiLFxuICAgIFwiMy4xMFwiLFxuICAgIFwiMi4xLjJcIixcbiAgICBcIjIwLjA0XCIsXG4gICk7XG4gIC8qKiBOZXVyb24gU0RLIDIuMjAuMCAqL1xuICBzdGF0aWMgcmVhZG9ubHkgU0RLXzJfMjBfMCA9IE5ldXJvbnhJbWFnZS5mcm9tU2RrVmVyc2lvbihcbiAgICBcIjIuMjAuMFwiLFxuICAgIFwiMy4xMFwiLFxuICAgIFwiMi4xLjJcIixcbiAgICBcIjIwLjA0XCIsXG4gICk7XG4gIC8qKiBOZXVyb24gU0RLIDIuMTkuMSAqL1xuICBzdGF0aWMgcmVhZG9ubHkgU0RLXzJfMTlfMSA9IE5ldXJvbnhJbWFnZS5mcm9tU2RrVmVyc2lvbihcbiAgICBcIjIuMTkuMVwiLFxuICAgIFwiMy4xMFwiLFxuICAgIFwiMi4xLjJcIixcbiAgICBcIjIwLjA0XCIsXG4gICk7XG4gIC8qKiBOZXVyb24gU0RLIDIuMTkuMCAqL1xuICBzdGF0aWMgcmVhZG9ubHkgU0RLXzJfMTlfMCA9IE5ldXJvbnhJbWFnZS5mcm9tU2RrVmVyc2lvbihcbiAgICBcIjIuMTkuMFwiLFxuICAgIFwiMy4xMFwiLFxuICAgIFwiMi4xLjJcIixcbiAgICBcIjIwLjA0XCIsXG4gICk7XG4gIC8qKiBOZXVyb24gU0RLIDIuMTguMiAqL1xuICBzdGF0aWMgcmVhZG9ubHkgU0RLXzJfMThfMiA9IE5ldXJvbnhJbWFnZS5mcm9tU2RrVmVyc2lvbihcbiAgICBcIjIuMTguMlwiLFxuICAgIFwiMy4xMFwiLFxuICAgIFwiMi4xLjJcIixcbiAgICBcIjIwLjA0XCIsXG4gICk7XG4gIC8qKiBOZXVyb24gU0RLIDIuMTguMSAqL1xuICBzdGF0aWMgcmVhZG9ubHkgU0RLXzJfMThfMSA9IE5ldXJvbnhJbWFnZS5mcm9tU2RrVmVyc2lvbihcbiAgICBcIjIuMTguMVwiLFxuICAgIFwiMy4xMFwiLFxuICAgIFwiMi4xLjJcIixcbiAgICBcIjIwLjA0XCIsXG4gICk7XG4gIC8qKiBOZXVyb24gU0RLIDIuMTguMCAqL1xuICBzdGF0aWMgcmVhZG9ubHkgU0RLXzJfMThfMCA9IE5ldXJvbnhJbWFnZS5mcm9tU2RrVmVyc2lvbihcbiAgICBcIjIuMTguMFwiLFxuICAgIFwiMy4xMFwiLFxuICAgIFwiMi4xLjJcIixcbiAgICBcIjIwLjA0XCIsXG4gICk7XG4gIC8qKiBMYXRlc3QgTmV1cm9uIFNESyAqL1xuICBzdGF0aWMgcmVhZG9ubHkgTEFURVNUID0gT2JqZWN0LmVudHJpZXMoTmV1cm9ueEltYWdlKVxuICAgIC5maWx0ZXIoKFtrXSkgPT4gay5zdGFydHNXaXRoKFwiU0RLX1wiKSlcbiAgICAuc29ydCgoW2sxXSwgW2syXSkgPT4gKGsxIDwgazIgPyAxIDogLTEpKVxuICAgIC5tYXAoKFtfLCB2XSkgPT4gdilbMF0gYXMgSU5ldXJvbnhJbWFnZTtcblxuICBzdGF0aWMgZnJvbVNka1ZlcnNpb24oXG4gICAgbmV1cm9ueFZlcnNpb246IHN0cmluZyxcbiAgICBweXRob25WZXJzaW9uOiBzdHJpbmcsXG4gICAgcHl0b3JjaFZlcnNpb246IHN0cmluZyxcbiAgICB1YnVudHVWZXJzaW9uOiBzdHJpbmcsXG4gICk6IElOZXVyb254SW1hZ2Uge1xuICAgIHJldHVybiB7XG4gICAgICBzZGtWZXJzaW9uOiBuZXVyb254VmVyc2lvbixcbiAgICAgIGltYWdlTmFtZTogdGhpcy5pbWFnZU5hbWUsXG4gICAgICBpbWFnZVRhZzogYCR7cHl0b3JjaFZlcnNpb259LW5ldXJvbngtcHkke3B5dGhvblZlcnNpb24ucmVwbGFjZShcIi5cIiwgXCJcIil9LXNkayR7bmV1cm9ueFZlcnNpb259LXVidW50dSR7dWJ1bnR1VmVyc2lvbn1gLFxuICAgIH07XG4gIH1cbiAgcHJvdGVjdGVkIHN0YXRpYyByZWFkb25seSBpbWFnZU5hbWU6IHN0cmluZztcbn1cblxuZXhwb3J0IGNsYXNzIFB5dG9yY2hUcmFpbmluZ05ldXJvbnhJbWFnZSBleHRlbmRzIE5ldXJvbnhJbWFnZSB7XG4gIHN0YXRpYyByZWFkb25seSBpbWFnZU5hbWUgPSBcInB1YmxpYy5lY3IuYXdzL25ldXJvbi9weXRvcmNoLXRyYWluaW5nLW5ldXJvbnhcIjtcbn1cbiJdfQ==