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