-
Notifications
You must be signed in to change notification settings - Fork 2.9k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Run Final_Jar_Testing_Linux_GPU in docker (#17533)
### Description 1. Create a package test image based on [RedHat UBI](https://www.redhat.com/en/blog/introducing-red-hat-universal-base-image) 2. Install TensorRT 8.6.1.6 in RedHat. (Ref. https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#maclearn-net-repo-install-rpm) 3. Run Final_Jar_Testing_Linux_GPU in docker (base image: nvidia/cuda:11.8.0-cudnn8-devel-ubi8) ### Motivation and Context [AB#18470](https://aiinfra.visualstudio.com/6a833879-cd9b-44a4-a9de-adc2d818f13c/_workitems/edit/18470) ### Verification https://dev.azure.com/aiinfra/Lotus/_build/results?buildId=354004&view=logs&j=8939b564-1402-57b5-92dc-510eba75e069&t=8939b564-1402-57b5-92dc-510eba75e069
- Loading branch information
Showing
4 changed files
with
82 additions
and
12 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
45 changes: 45 additions & 0 deletions
45
tools/ci_build/github/linux/docker/Dockerfile.package_ubi8_cuda11_8_tensorrt8_6
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,45 @@ | ||
# -------------------------------------------------------------- | ||
# Copyright (c) Microsoft Corporation. All rights reserved. | ||
# Licensed under the MIT License. | ||
# -------------------------------------------------------------- | ||
# Dockerfile to Test ONNX Runtime on UBI8 with CUDA 11.8 and TensorRT 8.6 | ||
|
||
# Build base image with required system packages | ||
FROM nvidia/cuda:11.8.0-cudnn8-devel-ubi8 AS base | ||
|
||
ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/src/tensorrt/bin:${PATH} | ||
|
||
RUN dnf install -y bash wget &&\ | ||
dnf clean dbcache | ||
|
||
# Install python3 | ||
RUN dnf install -y \ | ||
python3.8 \ | ||
python38-pip \ | ||
python38-wheel &&\ | ||
cd /usr/local/bin &&\ | ||
ln -s /usr/bin/python3 python3.8 &&\ | ||
ln -s /usr/bin/pip3 pip3.8; | ||
|
||
RUN pip3 install --upgrade pip | ||
RUN pip3 install setuptools>=41.0.0 | ||
|
||
# Install TensorRT | ||
RUN dnf install -y libnvinfer8 libnvonnxparsers8 libnvparsers8 libnvinfer-plugin8 libnvinfer-lean8 libnvinfer-vc-plugin8 libnvinfer-dispatch8 | ||
RUN v="8.6.1.6-1+cuda11.8" &&\ | ||
dnf downgrade -y libnvinfer8-${v} libnvinfer8-${v} libnvonnxparsers8-${v} libnvparsers8-${v} libnvinfer-plugin8-${v} libnvinfer-lean8-${v} libnvinfer-vc-plugin8-${v} libnvinfer-dispatch8-${v} &&\ | ||
dnf install -y dnf-plugin-versionlock &&\ | ||
dnf versionlock libnvinfer8 libnvonnxparsers8 libnvparsers8 libnvinfer-plugin8 libnvinfer-lean8 libnvinfer-vc-plugin8 libnvinfer-dispatch8 | ||
RUN dnf clean dbcache | ||
|
||
|
||
ADD scripts /tmp/scripts | ||
RUN cd /tmp/scripts && /tmp/scripts/install_dotnet.sh && /tmp/scripts/install_java.sh && rm -rf /tmp/scripts | ||
|
||
# Build final image from base. | ||
FROM base as final | ||
ARG BUILD_USER=onnxruntimedev | ||
ARG BUILD_UID=1000 | ||
RUN adduser --uid $BUILD_UID $BUILD_USER | ||
WORKDIR /home/$BUILD_USER | ||
USER $BUILD_USER |
12 changes: 12 additions & 0 deletions
12
tools/ci_build/github/linux/docker/scripts/install_java.sh
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,12 @@ | ||
#!/bin/bash | ||
set -e -x | ||
|
||
if [ -f /etc/redhat-release ]; then | ||
dnf install -y java-11-openjdk-devel \ | ||
&& dnf clean dbcache | ||
elif [ -f /etc/os-release ]; then | ||
apt-get update && apt-get install -y openjdk-11-jdk | ||
else | ||
echo "Unsupported OS" | ||
exit 1 | ||
fi |