migrate CI to python 3.12 and lint to 3.10 to match core CI (#180)

stack-info: PR: #180, branch: xmfan/stack/11

Author: xmfan

.github/workflows/lint.yml

@@ -18,7 +18,7 @@ jobs:
       - name: Setup Python
         uses: actions/setup-python@v2
         with:
-          python-version: '3.9'
+          python-version: '3.10'
       - name: Install deps
         run: |
           pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu

.github/workflows/test_cuda.yml

@@ -13,7 +13,7 @@ concurrency:
 
 jobs:
   test-cuda:
-    name: Test CUDA (cuda12.6-py3.10)
+    name: Test CUDA (cuda12.6-py3.12)
     uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
     strategy:
       fail-fast: true
@@ -30,6 +30,7 @@ jobs:
       gpu-arch-type: ${{ matrix.gpu-arch-type }}
       gpu-arch-version: ${{ matrix.gpu-arch-version }}
       submodules: recursive
+      python-version: "3.12"
       script: |
         pip install --quiet -r requirements-test.txt
         # For some reason the spec above isnt working

.github/workflows/test_torchtitan.yml

@@ -13,7 +13,7 @@ concurrency:
 
 jobs:
   test-torchtitan:
-    name: Test TorchTitan Integration (cuda12.6-py3.10)
+    name: Test TorchTitan Integration (cuda12.6-py3.12)
     uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
     strategy:
       fail-fast: true
@@ -30,6 +30,7 @@ jobs:
       gpu-arch-type: ${{ matrix.gpu-arch-type }}
       gpu-arch-version: ${{ matrix.gpu-arch-version }}
       submodules: recursive
+      python-version: "3.12"
       script: |
         pip install --quiet -r requirements-test.txt
         # For some reason the spec above isnt working