Update docgen.yml

narendasan · web-flow · commit b1f01b20612c · 2024-12-04T09:51:22.000-07:00
diff --git a/.github/workflows/docgen.yml b/.github/workflows/docgen.yml
@@ -17,9 +17,9 @@ jobs:
       image: docker.io/pytorch/manylinux2_28-builder:cuda12.6
       options: --gpus all
     env:
-      CUDA_HOME: /usr/local/cuda-12.4
-      VERSION_SUFFIX: cu124
-      CU_VERSION: cu124
+      CUDA_HOME: /usr/local/cuda-12.6
+      VERSION_SUFFIX: cu126
+      CU_VERSION: cu126
       CHANNEL: nightly
       CI_BUILD: 1
     steps:
