make python-version and container consistent python versions

databricks · Aug 8, 2024 · 4d6ce4b · 4d6ce4b
1 parent 3b342ff
commit 4d6ce4b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml
@@ -21,9 +21,9 @@ jobs:
       fail-fast: false
       matrix:
         include:
-        - name: "python3.9-pytorch2.3.1-gpus1"
+        - name: "python3.11-pytorch2.3.1-gpus1"
           gpu_num: 1
-          python_version: 3.9
+          python_version: 3.11
           container: mosaicml/pytorch:2.3.1_cu121-python3.11-ubuntu20.04
         - name: "python3.11-pytorch2.3.1-gpus2"
           gpu_num: 2