Merge branch 'dev' into saaketh/pr-cpu-image-fix

mosaicml · Jun 18, 2024 · 52bc9dc · 52bc9dc
2 parents a54e1d3 + 08da687
commit 52bc9dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/pr-cpu.yaml b/.github/workflows/pr-cpu.yaml
@@ -22,7 +22,7 @@ jobs:
           markers: not daily and not remote and not gpu and not doctest
           pytest_command: coverage run -m pytest
         - name: cpu-3.11-2.3
-          container: mosaicml/pytorch:2.3.1_cpu-python3.11-ubuntu20.04
+          container: mosaicml/pytorch:2.3.1_cu121-python3.11-ubuntu20.04
           markers: not daily and not remote and not gpu and not doctest
           pytest_command: coverage run -m pytest
         - name: cpu-doctest