From be6e7e12471691399e3211bf384198161b5e7cb9 Mon Sep 17 00:00:00 2001 From: Changming Sun Date: Thu, 12 Oct 2023 21:01:08 +0000 Subject: [PATCH 1/2] Add CUDA 12 docker file --- .../docker/Dockerfile.manylinux2_28_cuda11_8_tensorrt8_6 | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tools/ci_build/github/linux/docker/Dockerfile.manylinux2_28_cuda11_8_tensorrt8_6 b/tools/ci_build/github/linux/docker/Dockerfile.manylinux2_28_cuda11_8_tensorrt8_6 index 2c953a10cbf64..14dc827d59fe8 100644 --- a/tools/ci_build/github/linux/docker/Dockerfile.manylinux2_28_cuda11_8_tensorrt8_6 +++ b/tools/ci_build/github/linux/docker/Dockerfile.manylinux2_28_cuda11_8_tensorrt8_6 @@ -1,4 +1,4 @@ -ARG BASEIMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubi8 +ARG BASEIMAGE=nvidia/cuda:11.8.0-devel-ubi8 ARG POLICY=manylinux_2_28 ARG PLATFORM=x86_64 ARG DEVTOOLSET_ROOTPATH= @@ -163,10 +163,8 @@ CMD ["/bin/bash"] #Build manylinux2014 docker image end #Install TensorRT 8.6.1.6 -#RUN yum install -y wget RUN v="8.6.1.6-1.cuda11.8" &&\ - yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/cuda-rhel8.repo &&\ - yum -y install libnvinfer8-${v} libnvparsers8-${v} libnvonnxparsers8-${v} libnvinfer-plugin8-${v} libnvinfer-vc-plugin8-${v}\ + dnf -y install libcudnn8-devel libnvinfer8-${v} libnvparsers8-${v} libnvonnxparsers8-${v} libnvinfer-plugin8-${v} libnvinfer-vc-plugin8-${v}\ libnvinfer-devel-${v} libnvparsers-devel-${v} libnvonnxparsers-devel-${v} libnvinfer-plugin-devel-${v} libnvinfer-vc-plugin-devel-${v} libnvinfer-headers-devel-${v} libnvinfer-headers-plugin-devel-${v} ENV JAVA_HOME=/usr/lib/jvm/msopenjdk-11 #Add our own dependencies From 660dd6dd6f502d6b40378d4aab4b5d57f4163383 Mon Sep 17 00:00:00 2001 From: Changming Sun Date: Thu, 12 Oct 2023 22:43:08 +0000 Subject: [PATCH 2/2] update --- .../docker/Dockerfile.manylinux2_28_cuda11_8_tensorrt8_6 | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/ci_build/github/linux/docker/Dockerfile.manylinux2_28_cuda11_8_tensorrt8_6 b/tools/ci_build/github/linux/docker/Dockerfile.manylinux2_28_cuda11_8_tensorrt8_6 index 14dc827d59fe8..5618f77a0e838 100644 --- a/tools/ci_build/github/linux/docker/Dockerfile.manylinux2_28_cuda11_8_tensorrt8_6 +++ b/tools/ci_build/github/linux/docker/Dockerfile.manylinux2_28_cuda11_8_tensorrt8_6 @@ -163,8 +163,8 @@ CMD ["/bin/bash"] #Build manylinux2014 docker image end #Install TensorRT 8.6.1.6 -RUN v="8.6.1.6-1.cuda11.8" &&\ - dnf -y install libcudnn8-devel libnvinfer8-${v} libnvparsers8-${v} libnvonnxparsers8-${v} libnvinfer-plugin8-${v} libnvinfer-vc-plugin8-${v}\ +RUN v="8.6.1.6-1.cuda11.8" && CUDA_VERSION=$(nvcc --version | sed -n 's/^.*release \([0-9]\+\.[0-9]\+\).*$/\1/p') \ + && dnf -y install libcudnn8-devel-*cuda$CUDA_VERSION* libcudnn8-*cuda$CUDA_VERSION* libnvinfer8-${v} libnvparsers8-${v} libnvonnxparsers8-${v} libnvinfer-plugin8-${v} libnvinfer-vc-plugin8-${v}\ libnvinfer-devel-${v} libnvparsers-devel-${v} libnvonnxparsers-devel-${v} libnvinfer-plugin-devel-${v} libnvinfer-vc-plugin-devel-${v} libnvinfer-headers-devel-${v} libnvinfer-headers-plugin-devel-${v} ENV JAVA_HOME=/usr/lib/jvm/msopenjdk-11 #Add our own dependencies