mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-18 21:21:17 +00:00
6958f49dae
* Revert "Remove docstrigs if __ONNX_NO_DOC_STRINGS" (#4495) This reverts commit bb4d331fa7bf1fe8d68b1527dda56e4739c80800. * Bump version to 1.4.0 (#4496) * Create N-1 threads in intra-op pool, given main thread now active (#4493) Create N-1 threads in a thread pool when configured with intra-op parallelism of N. This ensures we have N active threads, given that the main thread also runs work. To avoid ambiguity on the value returned, rename ThreadPool::NumThreads method to ThreadPool::DegreeOfParallelism, and make corresponding updates in MLAS and operators. * Conditionally compile without std::is_trivially_copyable to satisfy old GCC versions. (#4510) * Adding CUDA arch flags for NVIDIA Jetson Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Added Dockerfile for Jetson and instructions to build wheel and image Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Removing guess about nvcc location Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Restoring pip3 setuptools install order Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Updated README with links and notes re NVIDIA Docker runtime Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Added mention of nvidia-docker Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Addressing code review comments Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> * Addressing code review comments Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com> Co-authored-by: Tiago Koji Castro Shibata <ticastro@microsoft.com> Co-authored-by: Dmitri Smirnov <yuslepukhin@users.noreply.github.com> Co-authored-by: Tim Harris <tiharr@microsoft.com> Co-authored-by: edgchen1 <18449977+edgchen1@users.noreply.github.com>
37 lines
1 KiB
Text
37 lines
1 KiB
Text
# syntax=docker/dockerfile:experimental
|
|
#
|
|
# This Dockerfile just installs pre-built ONNX Runtime wheel inside the image.
|
|
# Please make sure you have nvidia-runtime enabled in docker config and then build like:
|
|
#
|
|
# sudo -H DOCKER_BUILDKIT=1 nvidia-docker build --build-arg WHEEL_FILE=<path> -f Dockerfile.jetson
|
|
#
|
|
|
|
ARG BASE_IMAGE=nvcr.io/nvidia/l4t-base:r32.4.3
|
|
FROM ${BASE_IMAGE} as onnxruntime
|
|
|
|
ARG WHEEL_FILE
|
|
RUN echo "Building ONNX Runtime Docker image with ${WHEEL_FILE}..."
|
|
|
|
# Ensure apt-get won't prompt for selecting options
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
|
RUN apt update && \
|
|
apt install -y --no-install-recommends \
|
|
build-essential \
|
|
software-properties-common \
|
|
libopenblas-dev \
|
|
libpython3.6-dev \
|
|
python3-pip \
|
|
python3-dev \
|
|
cmake
|
|
|
|
RUN pip3 install --upgrade pip
|
|
RUN pip3 install setuptools
|
|
RUN pip3 install wheel pybind11 pytest
|
|
|
|
WORKDIR /onnxruntime
|
|
|
|
# copy previously built wheel into the container
|
|
COPY ${WHEEL_FILE} .
|
|
|
|
RUN basename ${WHEEL_FILE} | xargs pip3 install
|