diff --git a/.github/workflows/torch-wheel.yml b/.github/workflows/torch-wheel.yml
index e82202f..32f34af 100644
--- a/.github/workflows/torch-wheel.yml
+++ b/.github/workflows/torch-wheel.yml
@@ -82,7 +82,7 @@ jobs:
         mv ${{ env.CACHE_PATH }}/torch-wheel-new ${{ env.CACHE_PATH }}/torch-wheel
 
     - name: Attest
-      uses: actions/attest-build-provenance@v1
+      uses: actions/attest-build-provenance@v2
       if: ${{ inputs.push-to-registry }}
       with:
         subject-name: index.docker.io/${{ vars.DOCKERHUB_USERNAME }}/torch-wheels
diff --git a/.github/workflows/torchvision-wheel.yml b/.github/workflows/torchvision-wheel.yml
index 40d29c7..c026bdb 100644
--- a/.github/workflows/torchvision-wheel.yml
+++ b/.github/workflows/torchvision-wheel.yml
@@ -90,7 +90,7 @@ jobs:
         mv ${{ env.CACHE_PATH }}/torchvision-wheel-new ${{ env.CACHE_PATH }}/torchvision-wheel
 
     - name: Attest
-      uses: actions/attest-build-provenance@v1
+      uses: actions/attest-build-provenance@v2
       if: ${{ inputs.push-to-registry }}
       with:
         subject-name: index.docker.io/${{ vars.DOCKERHUB_USERNAME }}/torchvision-wheels