add cuda11

tools/dockerfiles/build_scripts/install_whl.sh

@@ -53,6 +53,9 @@ elif [[ $SERVING_VERSION == "0.6.0" ]]; then
     elif [[ "$RUN_ENV" == "cuda10.2" ]];then
         server_release="https://paddle-serving.bj.bcebos.com/test-dev/whl/paddle_serving_server_gpu-$SERVING_VERSION.post102-py3-none-any.whl"
         serving_bin="https://paddle-serving.bj.bcebos.com/test-dev/bin/serving-gpu-102-$SERVING_VERSION.tar.gz"
+    elif [[ "$RUN_ENV" == "cuda11" ]];then
+        server_release="https://paddle-serving.bj.bcebos.com/test-dev/whl/paddle_serving_server_gpu-$SERVING_VERSION.post11-py3-none-any.whl"
+        serving_bin="https://paddle-serving.bj.bcebos.com/test-dev/bin/serving-gpu-11-$SERVING_VERSION.tar.gz"
     fi
     client_release="https://paddle-serving.bj.bcebos.com/test-dev/whl/paddle_serving_client-$SERVING_VERSION-cp$CPYTHON-none-any.whl"
     app_release="https://paddle-serving.bj.bcebos.com/test-dev/whl/paddle_serving_app-$SERVING_VERSION-py3-none-any.whl"
@@ -88,6 +91,16 @@ elif [[ "$RUN_ENV" == "cuda10.2" ]];then
     echo "export SERVING_BIN=$PWD/serving_bin/serving">>/root/.bashrc
     rm -rf serving-gpu-102-${SERVING_VERSION}.tar.gz
     cd -
+elif [[ "$RUN_ENV" == "cuda11" ]];then
+    python$PYTHON_VERSION -m pip install $client_release $app_release $server_release
+    python$PYTHON_VERSION -m pip install paddlepaddle-gpu==${PADDLE_VERSION}
+    cd /usr/local/
+    wget $serving_bin
+    tar xf serving-gpu-11-${SERVING_VERSION}.tar.gz
+    mv $PWD/serving-gpu-11-${SERVING_VERSION} $PWD/serving_bin
+    echo "export SERVING_BIN=$PWD/serving_bin/serving">>/root/.bashrc
+    rm -rf serving-gpu-11-${SERVING_VERSION}.tar.gz
+    cd -
 fi