chore: Update cmd-shell image to python3.11 (#656)

awslabs · Sep 18, 2024 · 8af24f6 · 8af24f6
1 parent 5b1788b
commit 8af24f6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/ai-ml/trainium-inferentia/examples/docker/Dockerfile-cmd-shell b/ai-ml/trainium-inferentia/examples/docker/Dockerfile-cmd-shell
@@ -0,0 +1,7 @@
+FROM public.ecr.aws/amazonlinux/amazonlinux:2023
+
+RUN yum update -y && yum install -y awscli-2
+
+RUN python3 -m ensurepip --upgrade
+
+RUN pip3 install -U "huggingface_hub[cli]"
diff --git a/ai-ml/trainium-inferentia/fsx-for-lustre.tf b/ai-ml/trainium-inferentia/fsx-for-lustre.tf
@@ -101,7 +101,7 @@ metadata:
 spec:
   containers:
   - name: app
-    image: public.ecr.aws/aws-cli/aws-cli:2.13.1
+    image: public.ecr.aws/data-on-eks/cmd-shell:py3-awscli2
     command: ["/bin/sh", "-c"]
     args: ["while true; do sleep 30; done"]
     volumeMounts: