saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix TRT dockerfile documentation https://github.com/microsoft/onnxruntime/issues/14556 (#14600)

Browse source 
address https://github.com/microsoft/onnxruntime/issues/14556
This commit is contained in:
George Wu 2023-03-01 07:02:42 -08:00 committed by GitHub
parent b7fde84341
commit 6044312a43
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
dockerfiles/Dockerfile.tensorrt
View file

@ -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

6
dockerfiles/README.md
View file

@ -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