[DevOps] update dockerfiles.

jrpizani · Oct 9, 2023 · 6ed0e89 · 6ed0e89
1 parent 85ee8b5
commit 6ed0e89
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/devops/dockerfile/device-image/Dockerfile-Test b/devops/dockerfile/device-image/Dockerfile-Test
@@ -1,6 +1,6 @@
 ARG VERSION=test
 ARG IS_BUILDING_GPU_IMAGE=0
-ARG BASE_IMAGE=public.ecr.aws/x6k8q1x9/fedml-device-image:base
+ARG BASE_IMAGE=fedml/fedml-device-image:base
 FROM ${BASE_IMAGE}
 
 ADD ./devops/scripts/runner.sh ./fedml/runner.sh
@@ -15,9 +15,9 @@ RUN pip3 install -r ./fedml/requirements.txt
 COPY ./python ./fedml/fedml-pip
 WORKDIR ./fedml/fedml-pip
 RUN pip3 install -e ./
-RUN pip3 install -e '.[tensorflow]'
-RUN pip3 install -e '.[jax]'
-RUN pip3 install -e '.[mxnet]'
+#RUN pip3 install -e '.[tensorflow]'
+#RUN pip3 install -e '.[jax]'
+#RUN pip3 install -e '.[mxnet]'
 RUN pip3 install MNN==1.1.6
 
 WORKDIR /fedml

diff --git a/devops/dockerfile/server-agent/Dockerfile-Test b/devops/dockerfile/server-agent/Dockerfile-Test
@@ -15,9 +15,9 @@ RUN pip3 install -r ./fedml/requirements.txt
 COPY ./python ./fedml/fedml-pip
 WORKDIR ./fedml/fedml-pip
 RUN pip3 install -e ./
-RUN pip3 install -e '.[tensorflow]'
-RUN pip3 install -e '.[jax]'
-RUN pip3 install -e '.[mxnet]'
+#RUN pip3 install -e '.[tensorflow]'
+#RUN pip3 install -e '.[jax]'
+#RUN pip3 install -e '.[mxnet]'
 
 WORKDIR /fedml