[DevOps] update the Dockerfile for model inference.

fedml-alex · fedml-alex · commit 62f8760665e2 · 2023-03-31T14:32:48.000+08:00
diff --git a/devops/dockerfile/model-inference-ingress/Dockerfile b/devops/dockerfile/model-inference-ingress/Dockerfile
@@ -1,6 +1,6 @@
 ARG VERSION=dev
 ARG BASE_IMAGE=public.ecr.aws/x6k8q1x9/fedml-device-image:base
-ARG FEDML_PIP_HOME=/usr/local/lib/python3.7/dist-packages/fedml
+ARG FEDML_PIP_HOME=/usr/local/lib/python3.8/dist-packages/fedml
 FROM ${BASE_IMAGE}
 
 ADD ./devops/scripts/runner.sh ./fedml/runner.sh
diff --git a/devops/dockerfile/model-premise-master/Dockerfile b/devops/dockerfile/model-premise-master/Dockerfile
@@ -1,6 +1,6 @@
 ARG VERSION=dev
 ARG BASE_IMAGE=public.ecr.aws/x6k8q1x9/fedml-device-image:base
-ARG FEDML_PIP_HOME=/usr/local/lib/python3.7/dist-packages/fedml
+ARG FEDML_PIP_HOME=/usr/local/lib/python3.8/dist-packages/fedml
 FROM ${BASE_IMAGE}
 
 ADD ./devops/scripts/runner.sh ./fedml/runner.sh
diff --git a/devops/dockerfile/model-premise-slave/Dockerfile b/devops/dockerfile/model-premise-slave/Dockerfile
@@ -1,7 +1,7 @@
 ARG VERSION=dev
 ARG IS_BUILDING_GPU_IMAGE=0
 ARG BASE_IMAGE=public.ecr.aws/x6k8q1x9/fedml-device-image:base
-ARG FEDML_PIP_HOME=/usr/local/lib/python3.7/dist-packages/fedml
+ARG FEDML_PIP_HOME=/usr/local/lib/python3.8/dist-packages/fedml
 FROM ${BASE_IMAGE}
 
 ADD ./devops/scripts/runner.sh ./fedml/runner.sh
