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