From aa2f2cab7c67493d52455ef3a54f48516c2954e4 Mon Sep 17 00:00:00 2001
From: Xin Yang <105740670+xyang16@users.noreply.github.com>
Date: Mon, 16 Sep 2024 17:05:37 -0700
Subject: [PATCH] [ci] Update tensorrt native build container (#3477)

---
 .github/workflows/native_jni_s3_tensorrt.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/native_jni_s3_tensorrt.yml b/.github/workflows/native_jni_s3_tensorrt.yml
index 5285e837ed7..fb74613fdcc 100644
--- a/.github/workflows/native_jni_s3_tensorrt.yml
+++ b/.github/workflows/native_jni_s3_tensorrt.yml
@@ -10,7 +10,7 @@ permissions:
 jobs:
   build-tensorrt-jni-linux:
     runs-on: ubuntu-latest
-    container: deepjavalibrary/ubuntu18.04:tensorrt-cuda116
+    container: nvcr.io/nvidia/tensorrt:22.07-py3
     steps:
       - name: Install Environment
         run: pip3 install awscli --upgrade