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

TensorRT dockerfile updates (#3016)

Browse source 
* change npy installation

* update trt base image version

* update build instructions for arm64/jetson
This commit is contained in:
George Wu 2020-02-13 14:34:00 -08:00 committed by GitHub
parent c9f18756b9
commit 584ba71485
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
BUILD.md
View file

@ -183,6 +183,10 @@ See more information on the TensorRT Execution Provider [here](./docs/execution_
Dockerfile instructions are available [here](./dockerfiles#tensorrt)
#### Jetson (ARM64 Builds)
See [instructions](https://github.com/microsoft/onnxruntime/issues/2684#issuecomment-568548387) for additional information and tips related to building Onnxruntime with TensorRT Execution Provider on Jetson platforms (TX1/TX2, Nano)
---
### DNNL and MKLML
@ -550,13 +554,13 @@ git clone --recursive https://github.com/Microsoft/onnxruntime
# Start the basic build
cd /code/onnxruntime
./build.sh --config MinSizeRel --arm --update --build
./build.sh --config MinSizeRel --update --build
# Build Shared Library
./build.sh --config MinSizeRel --arm --build_shared_lib
./build.sh --config MinSizeRel --build_shared_lib
# Build Python Bindings and Wheel
./build.sh --config MinSizeRel --arm --enable_pybind --build_wheel
./build.sh --config MinSizeRel --enable_pybind --build_wheel
# Build Output
ls -l /code/onnxruntime/build/Linux/MinSizeRel/*.so

2
dockerfiles/Dockerfile.tensorrt
View file

@ -5,7 +5,7 @@
# Dockerfile to run ONNXRuntime with TensorRT integration
# nVidia TensorRT Base Image
FROM nvcr.io/nvidia/tensorrt:19.09-py3
FROM nvcr.io/nvidia/tensorrt:20.01-py3
MAINTAINER Vinitra Swamy "viswamy@microsoft.com"
ARG ONNXRUNTIME_REPO=https://github.com/Microsoft/onnxruntime

3
dockerfiles/scripts/install_common_deps.sh
View file

@ -15,8 +15,7 @@ wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-4.5.11-Linux-x86_64.
rm ~/miniconda.sh
/opt/miniconda/bin/conda clean -ya
/opt/miniconda/bin/conda install -y numpy
/opt/miniconda/bin/conda clean -aqy
pip install numpy
rm -rf /opt/miniconda/pkgs
# Dependencies: cmake