update tensorboard and add explicit updates for awscli, boto3, and bo…

…tocore
aws · Jan 7, 2025 · f02da82 · f02da82
1 parent 3681ac6
commit f02da82
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/huggingface/pytorch/training/docker/2.1/py3/sdk2.20.0/Dockerfile.neuronx b/huggingface/pytorch/training/docker/2.1/py3/sdk2.20.0/Dockerfile.neuronx
@@ -23,20 +23,23 @@ RUN pip install --no-cache-dir \
 	datasets==${DATASETS_VERSION} \
 	optimum-neuron==${OPTIMUM_NEURON_VERSION} \
     peft \
-	"google-auth>=1.6.3,<2" \
 	gevent==${GEVENT_VERSION}
 
 # Pin numpy to version required by neuronx-cc
 # Update Pillow and urllib version to fix high and critical vulnerabilities
 RUN pip install -U \
+	"tensorboard>=2.11.0" \
 	"numpy>=1.24.3,<=1.25.2" \
 	"numba==0.58.1" \
 	"Pillow==10.3.0" \
 	"requests<2.32.0" \
 	Jinja2 \
 	mlflow \
 	tornado \
-	botocore==1.35.19 \
+	"awscli<2" \
+	boto3 \
+	botocore \
+	"google-auth>=1.6.3,<2" \
 	"urllib3>=1.26.17,<1.27"
 
 RUN apt-get update \