[ci] Update tensorrt native build container

deepjavalibrary · Sep 16, 2024 · 02f3052 · 02f3052
1 parent b185b0a
commit 02f3052
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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