diff --git a/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile b/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile
index fdd8e3ac51..4527c5b156 100644
--- a/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile
+++ b/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile
@@ -1,5 +1,9 @@
 FROM python:3.8
 
+# Input the feast version to install
+# This requires feast package to be available in pypi before building this image
+ARG VERSION
+
 RUN apt update && \
         apt install -y \
         jq \
@@ -7,8 +11,7 @@ RUN apt update && \
         build-essential
 
 RUN pip install pip --upgrade
-RUN pip install "feast[aws,gcp,snowflake,redis,go,mysql,postgres]"
-
+RUN pip install "feast[aws,gcp,snowflake,redis,go,mysql,postgres]==${VERSION}"
 
 RUN apt update
 RUN apt install -y -V ca-certificates lsb-release wget
diff --git a/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile.dev b/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile.dev
index 3fc1355d7a..015e3c7ee8 100644
--- a/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile.dev
+++ b/sdk/python/feast/infra/feature_servers/multicloud/Dockerfile.dev
@@ -1,5 +1,9 @@
 FROM python:3.8
 
+# Input the feast version to install
+# This requires feast package to be available in pypi before building this image
+ARG VERSION
+
 RUN apt update && \
         apt install -y \
         jq \
@@ -9,11 +13,11 @@ RUN apt update && \
 RUN pip install pip --upgrade
 COPY . .
 
-RUN pip install "feast[aws,gcp,snowflake,redis,go,mysql,postgres]"
+RUN pip install "feast[aws,gcp,snowflake,redis,go,mysql,postgres]==${VERSION}"
 
 RUN apt update
 RUN apt install -y -V ca-certificates lsb-release wget
 RUN wget https://apache.jfrog.io/artifactory/arrow/$(lsb_release --id --short | tr 'A-Z' 'a-z')/apache-arrow-apt-source-latest-$(lsb_release --codename --short).deb
 RUN apt install -y -V ./apache-arrow-apt-source-latest-$(lsb_release --codename --short).deb
 RUN apt update
-RUN apt -y install libarrow-dev
\ No newline at end of file
+RUN apt -y install libarrow-dev