diff --git a/docker/pytorch/inference/1.13.1/Dockerfile.neuron b/docker/pytorch/inference/1.13.1/Dockerfile.neuron index 8a1089c..da386ec 100644 --- a/docker/pytorch/inference/1.13.1/Dockerfile.neuron +++ b/docker/pytorch/inference/1.13.1/Dockerfile.neuron @@ -5,13 +5,13 @@ LABEL maintainer="Amazon AI" LABEL com.amazonaws.sagemaker.capabilities.accept-bind-to-port=true # Neuron SDK components version numbers -ARG NEURON_FRAMEWORK_VERSION=1.13.1.2.9.74.0 -ARG NEURON_CC_VERSION=1.22.0.0 -ARG NEURONX_TOOLS_VERSION=2.17.1.0 +ARG NEURON_FRAMEWORK_VERSION=1.13.1.2.10.12.0 +ARG NEURON_CC_VERSION=1.23.5.0 +ARG NEURONX_TOOLS_VERSION=2.18.3.0 ARG PYTHON=python3.10 ARG PYTHON_VERSION=3.10.12 -ARG TORCHSERVE_VERSION=0.10.0 +ARG TORCHSERVE_VERSION=0.11.0 ARG SM_TOOLKIT_VERSION=2.0.21 ARG MAMBA_VERSION=23.1.0-4 diff --git a/docker/pytorch/inference/1.13.1/Dockerfile.neuronx b/docker/pytorch/inference/1.13.1/Dockerfile.neuronx index 027c00d..ca4593a 100644 --- a/docker/pytorch/inference/1.13.1/Dockerfile.neuronx +++ b/docker/pytorch/inference/1.13.1/Dockerfile.neuronx @@ -5,17 +5,17 @@ LABEL maintainer="Amazon AI" LABEL com.amazonaws.sagemaker.capabilities.accept-bind-to-port=true # Neuron SDK components version numbers -ARG NEURONX_FRAMEWORK_VERSION=1.13.1.1.14.0 -ARG NEURONX_DISTRIBUTED_VERSION=0.7.0 -ARG NEURONX_CC_VERSION=2.13.72.0 -ARG NEURONX_TRANSFORMERS_VERSION=0.10.0.360 -ARG NEURONX_COLLECTIVES_LIB_VERSION=2.20.22.0-c101c322e -ARG NEURONX_RUNTIME_LIB_VERSION=2.20.22.0-1b3ca6425 -ARG NEURONX_TOOLS_VERSION=2.17.1.0 +ARG NEURONX_FRAMEWORK_VERSION=1.13.1.1.15.0 +ARG NEURONX_DISTRIBUTED_VERSION=0.8.0 +ARG NEURONX_CC_VERSION=2.14.213.0 +ARG NEURONX_TRANSFORMERS_VERSION=0.11.351 +ARG NEURONX_COLLECTIVES_LIB_VERSION=2.21.46.0-69b77134b +ARG NEURONX_RUNTIME_LIB_VERSION=2.21.41.0-fb1705f5f +ARG NEURONX_TOOLS_VERSION=2.18.3.0 ARG PYTHON=python3.10 ARG PYTHON_VERSION=3.10.12 -ARG TORCHSERVE_VERSION=0.10.0 +ARG TORCHSERVE_VERSION=0.11.0 ARG SM_TOOLKIT_VERSION=2.0.21 ARG MAMBA_VERSION=23.1.0-4 diff --git a/docker/pytorch/inference/2.1.2/Dockerfile.neuronx b/docker/pytorch/inference/2.1.2/Dockerfile.neuronx index 45ee4be..e8336ba 100644 --- a/docker/pytorch/inference/2.1.2/Dockerfile.neuronx +++ b/docker/pytorch/inference/2.1.2/Dockerfile.neuronx @@ -5,17 +5,17 @@ LABEL maintainer="Amazon AI" LABEL com.amazonaws.sagemaker.capabilities.accept-bind-to-port=true # Neuron SDK components version numbers -ARG NEURONX_DISTRIBUTED_VERSION=0.7.0 -ARG NEURONX_CC_VERSION=2.13.72.0 -ARG NEURONX_FRAMEWORK_VERSION=2.1.2.2.1.0 -ARG NEURONX_TRANSFORMERS_VERSION=0.10.0.360 -ARG NEURONX_COLLECTIVES_LIB_VERSION=2.20.22.0-c101c322e -ARG NEURONX_RUNTIME_LIB_VERSION=2.20.22.0-1b3ca6425 -ARG NEURONX_TOOLS_VERSION=2.17.1.0 +ARG NEURONX_DISTRIBUTED_VERSION=0.8.0 +ARG NEURONX_CC_VERSION=2.14.213.0 +ARG NEURONX_FRAMEWORK_VERSION=2.1.2.2.2.0 +ARG NEURONX_TRANSFORMERS_VERSION=0.11.351 +ARG NEURONX_COLLECTIVES_LIB_VERSION=2.21.46.0-69b77134b +ARG NEURONX_RUNTIME_LIB_VERSION=2.21.41.0-fb1705f5f +ARG NEURONX_TOOLS_VERSION=2.18.3.0 ARG PYTHON=python3.10 ARG PYTHON_VERSION=3.10.12 -ARG TORCHSERVE_VERSION=0.10.0 +ARG TORCHSERVE_VERSION=0.11.0 ARG SM_TOOLKIT_VERSION=2.0.21 ARG MAMBA_VERSION=23.1.0-4 diff --git a/docker/pytorch/training/1.13.1/Dockerfile.neuronx b/docker/pytorch/training/1.13.1/Dockerfile.neuronx index 734c8f5..3b1abe8 100644 --- a/docker/pytorch/training/1.13.1/Dockerfile.neuronx +++ b/docker/pytorch/training/1.13.1/Dockerfile.neuronx @@ -4,12 +4,12 @@ LABEL maintainer="Amazon AI" LABEL dlc_major_version="1" # Neuron SDK components version numbers -ARG NEURONX_FRAMEWORK_VERSION=1.13.1.1.14.0 -ARG NEURONX_DISTRIBUTED_VERSION=0.7.0 -ARG NEURONX_CC_VERSION=2.13.72.0 -ARG NEURONX_COLLECTIVES_LIB_VERSION=2.20.22.0-c101c322e -ARG NEURONX_RUNTIME_LIB_VERSION=2.20.22.0-1b3ca6425 -ARG NEURONX_TOOLS_VERSION=2.17.1.0 +ARG NEURONX_FRAMEWORK_VERSION=1.13.1.1.15.0 +ARG NEURONX_DISTRIBUTED_VERSION=0.8.0 +ARG NEURONX_CC_VERSION=2.14.213.0 +ARG NEURONX_COLLECTIVES_LIB_VERSION=2.21.46.0-69b77134b +ARG NEURONX_RUNTIME_LIB_VERSION=2.21.41.0-fb1705f5f +ARG NEURONX_TOOLS_VERSION=2.18.3.0 ARG PYTHON=python3.10 ARG PYTHON_VERSION=3.10.12 diff --git a/docker/pytorch/training/2.1.2/Dockerfile.neuronx b/docker/pytorch/training/2.1.2/Dockerfile.neuronx index 59f5094..bf93bbf 100644 --- a/docker/pytorch/training/2.1.2/Dockerfile.neuronx +++ b/docker/pytorch/training/2.1.2/Dockerfile.neuronx @@ -4,12 +4,12 @@ LABEL maintainer="Amazon AI" LABEL dlc_major_version="1" # Neuron SDK components version numbers -ARG NEURONX_DISTRIBUTED_VERSION=0.7.0 -ARG NEURONX_CC_VERSION=2.13.72.0 -ARG NEURONX_FRAMEWORK_VERSION=2.1.2.2.1.0 -ARG NEURONX_COLLECTIVES_LIB_VERSION=2.20.22.0-c101c322e -ARG NEURONX_RUNTIME_LIB_VERSION=2.20.22.0-1b3ca6425 -ARG NEURONX_TOOLS_VERSION=2.17.1.0 +ARG NEURONX_DISTRIBUTED_VERSION=0.8.0 +ARG NEURONX_CC_VERSION=2.14.213.0 +ARG NEURONX_FRAMEWORK_VERSION=2.1.2.2.2.0 +ARG NEURONX_COLLECTIVES_LIB_VERSION=2.21.46.0-69b77134b +ARG NEURONX_RUNTIME_LIB_VERSION=2.21.41.0-fb1705f5f +ARG NEURONX_TOOLS_VERSION=2.18.3.0 ARG PYTHON=python3.10 ARG PYTHON_VERSION=3.10.12