revert Jenkinsfile

NVIDIA · Feb 8, 2023 · d23b775 · d23b775
1 parent a3dcb84
commit d23b775
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Jenkinsfile b/Jenkinsfile
@@ -1,7 +1,7 @@
 pipeline {
   agent {
         docker {
-          image 'nemo_containers:23.01_apex_c3d575f2478cd379b3c2d81f41edde39791b5d92'
+          image 'nvcr.io/nvidia/pytorch:23.01-py3'
           args '--device=/dev/nvidia0 --gpus all --user 0:128 -v /home/TestData:/home/TestData -v $HOME/.cache:/root/.cache --shm-size=8g'
         }
   }
@@ -4510,3 +4510,4 @@ assert_frame_equal(training_curve, gt_curve, rtol=1e-3, atol=1e-3)"'''
     }
   }
 }
+