diff --git a/serving/Dockerfile b/serving/Dockerfile
old mode 100644
new mode 100755
index c2a7f5677..110f68a7e
--- a/serving/Dockerfile
+++ b/serving/Dockerfile
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 
-FROM nvidia/cuda:11.6.1-cudnn8-devel-ubuntu20.04
+FROM nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04
 
 ARG http_proxy
 ARG https_proxy
diff --git a/serving/scripts/build.sh b/serving/scripts/build.sh
old mode 100644
new mode 100755
index ffcd43e36..9e7fdecbe
--- a/serving/scripts/build.sh
+++ b/serving/scripts/build.sh
@@ -86,7 +86,7 @@ nvidia-docker run -i --rm --name ${docker_name} \
            -e "http_proxy=${http_proxy}" \
            -e "https_proxy=${https_proxy}" \
            -e "trt_version=${trt_version}"\
-           nvidia/cuda:11.6.1-cudnn8-devel-ubuntu20.04 \
+           nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04  \
            bash -c \
            'export https_proxy_tmp=${https_proxy}
             export http_proxy_tmp=${http_proxy}