diff --git a/.github/workflows/docs.yaml b/.github/workflows/docs.yaml
index feae0827..5a97fbb5 100644
--- a/.github/workflows/docs.yaml
+++ b/.github/workflows/docs.yaml
@@ -26,7 +26,7 @@ jobs:
           miniconda-version: "latest"
           activate-environment: test
           python-version: '3.10'
-          auto-activate: false
+          auto-activate: true
       - name: Verify conda environment
         shell: bash -l {0}
         run: |
@@ -36,9 +36,6 @@ jobs:
       - name: Update pip
         shell: bash -l {0}
         run: python -m pip install --upgrade pip
-      - name: Install pytorch
-        shell: bash -l {0}
-        run: pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
       - name: Install Dependencies
         shell: bash -l {0}
         run: |
@@ -46,6 +43,7 @@ jobs:
           conda install -y git
           conda install -y -c conda-forge glog==0.4.0 gflags fmt
           pip install cmake
+          pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
           pip install -r docs/requirements.txt
           export USE_NCCL=0
           export USE_NCCLX=0
@@ -56,7 +54,6 @@ jobs:
         working-directory: docs
         run: |
           set -eux
-          conda activate test
           make html SPHINXOPTS="-WT --keep-going" || echo "Build completed with warnings/errors"
       - name: Remove noindex meta tag from docs
         working-directory: docs/build/html