diff --git a/.github/workflows/native_jni_s3_pytorch_android.yml b/.github/workflows/native_jni_s3_pytorch_android.yml
index 0426e9dedda..3163f29850b 100644
--- a/.github/workflows/native_jni_s3_pytorch_android.yml
+++ b/.github/workflows/native_jni_s3_pytorch_android.yml
@@ -54,5 +54,5 @@ jobs:
           aws-region: us-east-2
       - name: Copy files to S3 with the AWS CLI
         run: |
-          aws s3 sync jnilib s3://djl-ai/publish/pytorch-1.7.1/jnilib
-          aws cloudfront create-invalidation --distribution-id E371VB8JQ6NRVY --paths "/pytorch-1.7.1/jnilib*"
+          aws s3 sync jnilib s3://djl-ai/publish/pytorch-1.8.1/jnilib
+          aws cloudfront create-invalidation --distribution-id E371VB8JQ6NRVY --paths "/pytorch-1.8.1/jnilib*"
diff --git a/.github/workflows/native_s3_pytorch_android.yml b/.github/workflows/native_s3_pytorch_android.yml
index 5fbabbd16ab..14372ac058f 100644
--- a/.github/workflows/native_s3_pytorch_android.yml
+++ b/.github/workflows/native_s3_pytorch_android.yml
@@ -10,8 +10,8 @@ jobs:
       matrix:
         format: ["armeabi-v7a", "arm64-v8a", "x86" ,"x86_64"]
     env:
-      PYTORCH_TAG_VERSION:    "v1.7.1"
-      PYTORCH_VERSION: "1.7.1"
+      PYTORCH_TAG_VERSION:    "v1.8.1"
+      PYTORCH_VERSION: "1.8.1"
       NDK_VERSION: "20.0.5594570"
     steps:
       - uses: actions/checkout@v1
@@ -26,7 +26,7 @@ jobs:
       - name: clone pytorch
         run: git clone https://github.com/pytorch/pytorch android_pytorch_tmp -b "${PYTORCH_TAG_VERSION}" --recursive
       - name: install Python Dependencies
-        run: pip install numpy ninja pyyaml mkl mkl-include setuptools cmake cffi
+        run: pip install numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing_extensions
       - name: Install NDK
         run: echo "y" | sudo ${ANDROID_HOME}/tools/bin/sdkmanager --install "ndk;${NDK_VERSION}"
       - name: build android