From 55bcdffd1bd02265fcb2b7d3d20aa44a29bc3b07 Mon Sep 17 00:00:00 2001 From: jiqing-feng Date: Wed, 18 Dec 2024 06:30:13 -0500 Subject: [PATCH] upgrade pytorch to 2.5.1 in gptq tests Signed-off-by: jiqing-feng --- .github/workflows/test_gptq.yml | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/.github/workflows/test_gptq.yml b/.github/workflows/test_gptq.yml index 9b04589c670..31a739a81f2 100644 --- a/.github/workflows/test_gptq.yml +++ b/.github/workflows/test_gptq.yml @@ -30,8 +30,7 @@ jobs: - name: Run tests uses: addnab/docker-run-action@v3 with: - image: pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime - # latest auto-gptq was built with pytorch 2.2 and cuda 12.1 + image: pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime options: | --rm --gpus all