diff --git a/scripts/docker_run.sh b/scripts/docker_run.sh
index 5549a4a1f7..ae1e4c5557 100644
--- a/scripts/docker_run.sh
+++ b/scripts/docker_run.sh
@@ -83,33 +83,61 @@ function construnct_version(){
 
 function construnct_version2(){
     if [[ ${cuda_version} == '11.2' ]];then
-        rm -rf commitid
-        wget https://paddle-qa.bj.bcebos.com/WheelBenchmark/linux-gpu-cuda11.2-cudnn8-mkl-gcc8.2-avx/commitid
-        image_commit_id=`cat commitid|grep commitID:|awk -F 'commitID:' '{print $2}'`
-        image_commit_id6=${image_commit_id:0:6}
-        echo "image_commit_id is: "${image_commit_id}"\n image_commit_id6 is : "${image_commit_id6}
-        version_temp=`tail -1 commitid|awk -F '+0800' '{print $1}' |awk -F 'commitIDtime:' '{print $2}'`  # 2022-04-01 21:18:53
-        version=`date -d  "${version_temp}" "+%Y.%m%d.%H%M%S"`   # 2022.0401.211853
-        PADDLE_VERSION=${version}.post118.develop.${image_commit_id6}
-        IMAGE_NAME=paddlepaddle_gpu-0.0.0.post11.2_cudnn8.1_develop-cp37-cp37m-linux_x86_64.whl
-        mkdir ${all_path}/images/${PADDLE_VERSION}
-        unset ${http_proxy} && unset ${https_proxy} 
-        wget https://paddle-qa.bj.bcebos.com/WheelBenchmark/linux-gpu-cuda11.2-cudnn8-mkl-gcc8.2-avx/${image_commit_id}/${IMAGE_NAME} -P ${all_path}/images/${PADDLE_VERSION}/
-                
-    else 
-        rm -rf commitid
-        wget https://paddle-qa.bj.bcebos.com/WheelBenchmark/linux-gpu-cuda10.1-cudnn7-mkl-gcc5.4-avx/commitid
-        image_commit_id=`cat commitid|grep commitID:|awk -F 'commitID:' '{print $2}'`
-        image_commit_id6=${image_commit_id:0:6}
-        echo "image_commit_id is: "${image_commit_id}"\n image_commit_id6 is : "${image_commit_id6}
-        version_temp=`tail -1 commitid|awk -F '+0800' '{print $1}' |awk -F 'commitIDtime:' '{print $2}'`  # 2022-04-01 21:18:53
-        version=`date -d  "${version_temp}" "+%Y.%m%d.%H%M%S"`   # 2022.0401.211853
-        PADDLE_VERSION=${version}.post101.develop.${image_commit_id6}
-        IMAGE_NAME=paddlepaddle_gpu-0.0.0.post10.1_cudnn7.6_develop-cp37-cp37m-linux_x86_64.whl
-        mkdir ${all_path}/images/${PADDLE_VERSION}
-        unset ${http_proxy} && unset ${https_proxy} 
-        wget https://paddle-qa.bj.bcebos.com/WheelBenchmark/linux-gpu-cuda10.1-cudnn7-mkl-gcc5.4-avx/${image_commit_id}/${IMAGE_NAME} -P ${all_path}/images/${PADDLE_VERSION}/
+        if [[ ${image_branch} == 'develop' ]];then
+            rm -rf commitid
+            wget https://paddle-qa.bj.bcebos.com/WheelBenchmark/linux-gpu-cuda11.2-cudnn8-mkl-gcc8.2-avx/commitid
+            image_commit_id=`cat commitid|grep commitID:|awk -F 'commitID:' '{print $2}'`
+            image_commit_id6=${image_commit_id:0:6}
+            echo "image_commit_id is: "${image_commit_id}"\n image_commit_id6 is : "${image_commit_id6}
+            version_temp=`tail -1 commitid|awk -F '+0800' '{print $1}' |awk -F 'commitIDtime:' '{print $2}'`  # 2022-04-01 21:18:53
+            version=`date -d  "${version_temp}" "+%Y.%m%d.%H%M%S"`   # 2022.0401.211853
+            PADDLE_VERSION=${version}.post118.develop.${image_commit_id6}
+            IMAGE_NAME=paddlepaddle_gpu-0.0.0.post11.2_cudnn8.1_develop-cp37-cp37m-linux_x86_64.whl
+            mkdir ${all_path}/images/${PADDLE_VERSION}
+            unset ${http_proxy} && unset ${https_proxy} 
+            wget https://paddle-qa.bj.bcebos.com/WheelBenchmark/linux-gpu-cuda11.2-cudnn8-mkl-gcc8.2-avx/${image_commit_id}/${IMAGE_NAME} -P ${all_path}/images/${PADDLE_VERSION}/
+        else   
+            rm -rf commitid
+            wget https://paddle-qa.bj.bcebos.com/WheelBenchmark/linux-gpu-cuda11.2-cudnn8-mkl-gcc8.2-avx/release/${image_branch}/commitid
+            image_commit_id=`cat commitid|grep commitID:|awk -F 'commitID:' '{print $2}'`
+            image_commit_id6=${image_commit_id:0:6}
+            echo "image_commit_id is: "${image_commit_id}"\n image_commit_id6 is : "${image_commit_id6}
+            version_temp=`tail -1 commitid|awk -F '+0800' '{print $1}' |awk -F 'commitIDtime:' '{print $2}'`  # 2022-04-01 21:18:53
+            version=`date -d  "${version_temp}" "+%Y.%m%d.%H%M%S"`   # 2022.0401.211853
+            PADDLE_VERSION=${version}.post118.develop.${image_commit_id6}
+            IMAGE_NAME=paddlepaddle_gpu-0.0.0.post11.2_cudnn8.1_${image_branch}-cp37-cp37m-linux_x86_64.whl
+            mkdir ${all_path}/images/${PADDLE_VERSION}
+            unset ${http_proxy} && unset ${https_proxy} 
+            wget https://paddle-qa.bj.bcebos.com/WheelBenchmark/linux-gpu-cuda11.2-cudnn8-mkl-gcc8.2-avx/release/${image_branch}/${image_commit_id}/${IMAGE_NAME} -P ${all_path}/images/${PADDLE_VERSION}/
+    elif [[ ${cuda_version} == '10.1' ]];then
+        if [[ ${image_branch} == 'develop' ]];then 
+            rm -rf commitid
+            wget https://paddle-qa.bj.bcebos.com/WheelBenchmark/linux-gpu-cuda10.1-cudnn7-mkl-gcc5.4-avx/commitid
+            image_commit_id=`cat commitid|grep commitID:|awk -F 'commitID:' '{print $2}'`
+            image_commit_id6=${image_commit_id:0:6}
+            echo "image_commit_id is: "${image_commit_id}"\n image_commit_id6 is : "${image_commit_id6}
+            version_temp=`tail -1 commitid|awk -F '+0800' '{print $1}' |awk -F 'commitIDtime:' '{print $2}'`  # 2022-04-01 21:18:53
+            version=`date -d  "${version_temp}" "+%Y.%m%d.%H%M%S"`   # 2022.0401.211853
+            PADDLE_VERSION=${version}.post101.develop.${image_commit_id6}
+            IMAGE_NAME=paddlepaddle_gpu-0.0.0.post10.1_cudnn7.6_develop-cp37-cp37m-linux_x86_64.whl
+            mkdir ${all_path}/images/${PADDLE_VERSION}
+            unset ${http_proxy} && unset ${https_proxy} 
+            wget https://paddle-qa.bj.bcebos.com/WheelBenchmark/linux-gpu-cuda10.1-cudnn7-mkl-gcc5.4-avx/${image_commit_id}/${IMAGE_NAME} -P ${all_path}/images/${PADDLE_VERSION}/
+        else 
+            rm -rf commitid
+            wget https://paddle-qa.bj.bcebos.com/WheelBenchmark/linux-gpu-cuda10.1-cudnn7-mkl-gcc5.4-avx/release/${image_branch}/commitid
+            image_commit_id=`cat commitid|grep commitID:|awk -F 'commitID:' '{print $2}'`
+            image_commit_id6=${image_commit_id:0:6}
+            echo "image_commit_id is: "${image_commit_id}"\n image_commit_id6 is : "${image_commit_id6}
+            version_temp=`tail -1 commitid|awk -F '+0800' '{print $1}' |awk -F 'commitIDtime:' '{print $2}'`  # 2022-04-01 21:18:53
+            version=`date -d  "${version_temp}" "+%Y.%m%d.%H%M%S"`   # 2022.0401.211853
+            PADDLE_VERSION=${version}.post101.develop.${image_commit_id6}
+            IMAGE_NAME=paddlepaddle_gpu-0.0.0.post10.1_cudnn7.6_${image_branch}-cp37-cp37m-linux_x86_64.whl
+            mkdir ${all_path}/images/${PADDLE_VERSION}
+            unset ${http_proxy} && unset ${https_proxy} 
+            wget https://paddle-qa.bj.bcebos.com/WheelBenchmark/linux-gpu-cuda10.1-cudnn7-mkl-gcc5.4-avx/release/${image_branch}/${image_commit_id}/${IMAGE_NAME} -P ${all_path}/images/${PADDLE_VERSION}/
         
+        fi
     echo "----------------- PADDLE_VERSION is: ${PADDLE_VERSION}"
     echo "----------------- IMAGE_NAME is: ${IMAGE_NAME}"
     fi