From 6044312a434987bdb2ddfd7fafbf294955f1693d Mon Sep 17 00:00:00 2001 From: George Wu Date: Wed, 1 Mar 2023 07:02:42 -0800 Subject: [PATCH] fix TRT dockerfile documentation https://github.com/microsoft/onnxruntime/issues/14556 (#14600) address https://github.com/microsoft/onnxruntime/issues/14556 --- dockerfiles/Dockerfile.tensorrt | 2 +- dockerfiles/README.md | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/dockerfiles/Dockerfile.tensorrt b/dockerfiles/Dockerfile.tensorrt index f1202ca57b..e2f21a4def 100644 --- a/dockerfiles/Dockerfile.tensorrt +++ b/dockerfiles/Dockerfile.tensorrt @@ -5,7 +5,7 @@ # Dockerfile to run ONNXRuntime with TensorRT integration # nVidia TensorRT Base Image -ARG TRT_CONTAINER_VERSION=21.12 +ARG TRT_CONTAINER_VERSION=22.12 FROM nvcr.io/nvidia/tensorrt:${TRT_CONTAINER_VERSION}-py3 ARG ONNXRUNTIME_REPO=https://github.com/Microsoft/onnxruntime diff --git a/dockerfiles/README.md b/dockerfiles/README.md index c198172e8c..41017e3a88 100644 --- a/dockerfiles/README.md +++ b/dockerfiles/README.md @@ -63,7 +63,7 @@ git submodule update --init ``` ## TensorRT -**Ubuntu 18.04, CUDA 11.0, TensorRT 7.1.3.4** +**Ubuntu 20.04, CUDA 11.8, TensorRT 8.5.1** 1. Update submodules ``` @@ -78,7 +78,9 @@ git submodule update --init 3. Run the Docker image ``` - docker run -it onnxruntime-trt + docker run --gpus all -it onnxruntime-trt + or + nvidia-docker run -it onnxruntime-trt ``` ## OpenVINO