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"