[NVIDIA] bump github actions (Dao-AILab#1996)

* Update GitHub Actions to use checkout@v5 and setup-python@v6; enhance compute capability support

* revert changes

* revert

* Update publish.yml

* Update publish.yml

* Update publish.yml

* Update publish.yml

* cuda-toolkit@v0.2.29

Author: johnnynunez

.github/workflows/_build.yml

@@ -43,7 +43,7 @@ jobs:
     name: Build wheel (${{ inputs.release-version }}-${{ inputs.python-version }}-${{ inputs.cuda-version }}-${{ inputs.torch-version }}-${{ inputs.cxx11_abi }})
     steps:
       - name: Checkout
-        uses: actions/checkout@v4
+        uses: actions/checkout@v5
         with:
           ref: ${{ inputs.release-version }}
           submodules: recursive
@@ -77,7 +77,7 @@ jobs:
 
       - name: Install CUDA ${{ inputs.cuda-version }}
         if: ${{ inputs.cuda-version != 'cpu' }}
-        uses: Jimver/cuda-toolkit@v0.2.27
+        uses: Jimver/cuda-toolkit@v0.2.29
         id: cuda-toolkit
         with:
           cuda: ${{ inputs.cuda-version }}

.github/workflows/pre-commit.yaml

@@ -22,10 +22,10 @@ jobs:
   pre-commit:
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
 
       - name: Set up Python
-        uses: actions/setup-python@v5
+        uses: actions/setup-python@v6
         with:
           python-version: '3.11'
 

.github/workflows/publish.yml

@@ -41,17 +41,20 @@ jobs:
         # Using ubuntu-22.04 instead of 24.04 for more compatibility (glibc). Ideally we'd use the
         # manylinux docker image, but I haven't figured out how to install CUDA on manylinux.
         os: [ubuntu-22.04, ubuntu-22.04-arm]
-        python-version: ["3.9", "3.10", "3.11", "3.12", "3.13"]
-        torch-version: ["2.4.0", "2.5.1", "2.6.0", "2.7.1", "2.8.0"]
+        python-version: ["3.10", "3.11", "3.12", "3.13"]
+        torch-version: ["2.5.1", "2.6.0", "2.7.1", "2.8.0", "2.9.1"]
         cuda-version: ["12.9.1"]
         # We need separate wheels that either uses C++11 ABI (-D_GLIBCXX_USE_CXX11_ABI) or not.
         # Pytorch wheels currently don't use it, but nvcr images have Pytorch compiled with C++11 ABI.
         # Without this we get import error (undefined symbol: _ZN3c105ErrorC2ENS_14SourceLocationESs)
         # when building without C++11 ABI and using it on nvcr images.
         cxx11_abi: ["FALSE", "TRUE"]
         include:
-            - torch-version: "2.9.0.dev20250904"
-              cuda-version: "13.0.0"
+            - torch-version: "2.9.1"
+              cuda-version: "13.0.2"
+              python-version: "3.14" 
+            - torch-version: "2.10.0.dev20251108"
+              cuda-version: "13.0.2"
         exclude:
           # see https://github.com/pytorch/pytorch/blob/main/RELEASE.md#release-compatibility-matrix
           # Pytorch < 2.5 does not support Python 3.13
@@ -72,8 +75,8 @@ jobs:
     needs: [build_wheels]
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
-      - uses: actions/setup-python@v5
+      - uses: actions/checkout@v5
+      - uses: actions/setup-python@v6
         with:
           python-version: "3.10"
       - name: Install dependencies