Use new images in conda tests

Lightning-AI · Jul 25, 2022 · 8a531ad · 8a531ad
1 parent 6b23771
commit 8a531ad
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/.github/workflows/ci-pytorch_test-conda.yml b/.github/workflows/ci-pytorch_test-conda.yml
@@ -18,16 +18,16 @@ defaults:
 jobs:
   conda:
     runs-on: ubuntu-20.04
-    container: pytorchlightning/pytorch_lightning:base-conda-py${{ matrix.python-version }}-torch${{ matrix.pytorch-version }}
+    container: pytorchlightning/pytorch_lightning:base-conda-py${{ matrix.python-version }}-torch${{ matrix.pytorch-version }}-cuda${{ matrix.cuda-version }}
     strategy:
       fail-fast: false
       matrix:
         # nightly: add when there's a release candidate
         include:
-          - {python-version: "3.8", pytorch-version: "1.9"}
-          - {python-version: "3.8", pytorch-version: "1.10"}
-          - {python-version: "3.9", pytorch-version: "1.11"}
-          - {python-version: "3.9", pytorch-version: "1.12"}
+          - {python-version: "3.8", pytorch-version: "1.9", cuda-version: "11.1"}
+          - {python-version: "3.8", pytorch-version: "1.10", cuda-version: "11.1"}
+          - {python-version: "3.9", pytorch-version: "1.11", cuda-version: "11.3.1"}
+          - {python-version: "3.9", pytorch-version: "1.12", cuda-version: "11.3.1"}
 
     timeout-minutes: 30