From 1c9a0ed20799320509cab25ab5cee3eb6732e532 Mon Sep 17 00:00:00 2001 From: Charles Tang Date: Thu, 14 Nov 2024 10:28:46 -0800 Subject: [PATCH] Fix composer gpu daily test to use torch 2.5.1 (#3712) Co-authored-by: Chuck Tang --- .github/workflows/pr-gpu.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml index 8a8a53f3c6..964c2695c7 100644 --- a/.github/workflows/pr-gpu.yaml +++ b/.github/workflows/pr-gpu.yaml @@ -45,7 +45,7 @@ jobs: matrix: include: - name: gpu-3.11-2.5-2 - container: mosaicml/pytorch:2.5.0_cu124-python3.11-ubuntu20.04 + container: mosaicml/pytorch:2.5.1_cu124-python3.11-ubuntu20.04 markers: not daily and not remote and gpu and (doctest or not doctest) pytest_command: coverage run -m pytest composer_package_name: mosaicml @@ -75,7 +75,7 @@ jobs: matrix: include: - name: gpu-3.11-2.5-4 - container: mosaicml/pytorch:2.5.0_cu124-python3.11-ubuntu20.04 + container: mosaicml/pytorch:2.5.1_cu124-python3.11-ubuntu20.04 markers: not daily and not remote and gpu and (doctest or not doctest) pytest_command: coverage run -m pytest composer_package_name: mosaicml