Fix github action CI for GPU (#625)

Summary: Pull Request resolved: #625 Differential Revision: D53292685
pytorch · Feb 1, 2024 · 3e26dff · 3e26dff
1 parent 4453702
commit 3e26dff
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/ci_gpu.yml b/.github/workflows/ci_gpu.yml
@@ -8,6 +8,7 @@ on:
     branches:
       - main
 
+jobs:
   unittest_multi_gpu:
     runs-on: linux.4xlarge.nvidia.gpu
     steps:
@@ -17,7 +18,7 @@ on:
       - name: Set up Python
         uses: actions/setup-python@v2
         with:
-          python-version: 3.9
+          python-version: '3.11'
 
       - name: Install dependencies
         run: |