[fix] fix image to run e2e test

ricosjp · Jan 7, 2025 · 9406002 · 9406002
1 parent 5b68dd4
commit 9406002
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
@@ -47,6 +47,7 @@ gpu_test:
     - gpu
 
 e2e_test:
+  image: pytorch/pytorch:2.5.0-cuda12.4-cudnn9-runtime
   stage: test
   script:
     - make e2e_test