diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
index 51217790e..ae0de2f09 100644
--- a/.github/workflows/docs.yml
+++ b/.github/workflows/docs.yml
@@ -58,7 +58,7 @@ jobs:
     - name: Install PyTorch
       shell: bash
       run: |
-        conda install -n build_binary --yes pytorch cpuonly -c pytorch-nightly
+        conda install -n build_binary pip install torch --index-url https://download.pytorch.org/whl/nightly/cpu
     - name: Install fbgemm
       run: |
         conda run -n build_binary pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cpu
diff --git a/.github/workflows/unittest_ci.yml b/.github/workflows/unittest_ci.yml
index 0b3fe0638..1f0b18dcd 100644
--- a/.github/workflows/unittest_ci.yml
+++ b/.github/workflows/unittest_ci.yml
@@ -63,7 +63,7 @@ jobs:
             python-version: '3.12'
             python-tag: "py312"
             cuda-tag: "cu124"
-    uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
+    uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
     with:
       runner: ${{ matrix.os }}
       timeout: 30
diff --git a/.github/workflows/unittest_ci_cpu.yml b/.github/workflows/unittest_ci_cpu.yml
index 1efe64178..4c92e6027 100644
--- a/.github/workflows/unittest_ci_cpu.yml
+++ b/.github/workflows/unittest_ci_cpu.yml
@@ -35,7 +35,7 @@ jobs:
             - os: linux.2xlarge
               python-version: '3.12'
               python-tag: "py312"
-    uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
+    uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
     with:
       runner: ${{ matrix.os }}
       timeout: 15