Doc build pipeline issue fix (#2985)

pytorch · Jul 5, 2024 · 9841392 · 9841392
1 parent 624ec8a
commit 9841392
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/docgen.yml b/.github/workflows/docgen.yml
@@ -19,6 +19,8 @@ jobs:
     env:
       CUDA_HOME: /usr/local/cuda-12.4
       VERSION_SUFFIX: cu124
+      CU_VERSION: cu124
+      CHANNEL: nightly
       CI_BUILD: 1
     steps:
       - uses: actions/checkout@v3