try cuda 12.8

Author: yf225

.github/matrix.json

@@ -56,8 +56,8 @@
       "runner": "linux.aws.h100.4",
       "python-version": "3.12",
       "ref-eager": false,
-      "image": "nvidia/cuda:13.0.1-devel-ubuntu24.04",
-      "runtime-version": "cu130",
+      "image": "nvidia/cuda:12.8.1-devel-ubuntu24.04",
+      "runtime-version": "cu128",
       "container-options": "--gpus all",
       "pytorch-version": "pytorch-nightly",
       "alias": "h100-distributed"

.github/workflows/test.yml

@@ -56,12 +56,6 @@ jobs:
       - name: Check out code
         uses: actions/checkout@v5
 
-      - name: Install NVIDIA driver
-        if: contains(matrix.alias, 'distributed')
-        uses: ./.github/actions/setup-nvidia-helion
-        with:
-          driver-version: 580.82.07
-
       - name: Run NVIDIA command
         if: startsWith(matrix.image, 'nvidia')
         run: |