From 7e2b79b445451b23752581b68bbd54a1bcbe1633 Mon Sep 17 00:00:00 2001 From: Raphael Glon Date: Mon, 10 Mar 2025 17:39:45 +0100 Subject: [PATCH] fixing neuronx dependencies + optimum upgrade optimum upgrade needed, related to unpinned requirement issue: https://github.com/aws-neuron/aws-neuron-sdk/issues/1012 Signed-off-by: Raphael Glon --- dockerfiles/pytorch/Dockerfile.inf2 | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/dockerfiles/pytorch/Dockerfile.inf2 b/dockerfiles/pytorch/Dockerfile.inf2 index 38957055..d84c43ce 100644 --- a/dockerfiles/pytorch/Dockerfile.inf2 +++ b/dockerfiles/pytorch/Dockerfile.inf2 @@ -8,16 +8,17 @@ ARG PYTHON_VERSION=3.10.12 ARG MAMBA_VERSION=23.1.0-4 # Neuron SDK components version numbers -ARG NEURONX_FRAMEWORK_VERSION=2.1.2.2.1.0 -ARG NEURONX_DISTRIBUTED_VERSION=0.7.0 -ARG NEURONX_CC_VERSION=2.13.66.0 -ARG NEURONX_TRANSFORMERS_VERSION=0.10.0.21 -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=2.1.2.2.1.0 +# ARG NEURONX_DISTRIBUTED_VERSION=0.7.0 +# ARG NEURONX_CC_VERSION=2.13.66.0 +ARG NEURONX_TRANSFORMERS_VERSION=0.12.313 +ARG NEURONX_COLLECTIVES_LIB_VERSION=2.22.33.0-d2128d1aa +ARG NEURONX_RUNTIME_LIB_VERSION=2.22.19.0-5856c0b42 +ARG NEURONX_TOOLS_VERSION=2.19.0.0 + # HF ARGS -ARG OPTIMUM_NEURON_VERSION=0.0.23 +ARG OPTIMUM_NEURON_VERSION=0.0.28 # See http://bugs.python.org/issue19846 ENV LANG C.UTF-8