From d23b7757e0f935dacde2840f234193c632a2b3be Mon Sep 17 00:00:00 2001 From: Sangkug Lym Date: Tue, 7 Feb 2023 16:04:40 -0800 Subject: [PATCH] revert Jenkinsfile --- Jenkinsfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/Jenkinsfile b/Jenkinsfile index f36839598941..37679ac338e7 100644 --- 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)"''' } } } +