| ARG RELEASE |
| ARG LAUNCHPAD_BUILD_ARCH |
| LABEL org.opencontainers.image.ref.name=ubuntu |
| LABEL org.opencontainers.image.version=22.04 |
| ADD file:2b3b5254f38a790d40e31cb26155609f7fc99ef7bc99eae1e0d67fa9ae605f77 in / |
| CMD ["/bin/bash"] |
| RUN /bin/sh -c export DEBIAN_FRONTEND=noninteractive && apt-get update && apt-get install -y --no-install-recommends apt-utils build-essential ca-certificates curl libncurses5 libncursesw5 patch wget rsync unzip jq gnupg libtcmalloc-minimal4 # buildkit |
| ARG CUDA_VERSION |
| ARG CUDA_DRIVER_VERSION |
| ARG JETPACK_HOST_MOUNTS |
| ENV CUDA_VERSION=12.3.2.001 CUDA_DRIVER_VERSION=545.23.08 CUDA_CACHE_DISABLE=1 NVIDIA_REQUIRE_JETPACK_HOST_MOUNTS= |
| RUN |3 CUDA_VERSION=12.3.2.001 CUDA_DRIVER_VERSION=545.23.08 JETPACK_HOST_MOUNTS= /bin/sh -c if [ -n "${JETPACK_HOST_MOUNTS}" ]; then echo "/usr/lib/aarch64-linux-gnu/tegra" > /etc/ld.so.conf.d/nvidia-tegra.conf && echo "/usr/lib/aarch64-linux-gnu/tegra-egl" >> /etc/ld.so.conf.d/nvidia-tegra.conf; fi # buildkit |
| RUN |3 CUDA_VERSION=12.3.2.001 CUDA_DRIVER_VERSION=545.23.08 JETPACK_HOST_MOUNTS= /bin/sh -c /nvidia/build-scripts/installCUDA.sh # buildkit |
| RUN |3 CUDA_VERSION=12.3.2.001 CUDA_DRIVER_VERSION=545.23.08 JETPACK_HOST_MOUNTS= /bin/sh -c cp -vprd /nvidia/. / && patch -p0 < /etc/startup_scripts.patch && rm -f /etc/startup_scripts.patch # buildkit |
| ENV _CUDA_COMPAT_PATH=/usr/local/cuda/compat ENV=/etc/shinit_v2 BASH_ENV=/etc/bash.bashrc SHELL=/bin/bash NVIDIA_REQUIRE_CUDA=cuda>=9.0 |
| LABEL com.nvidia.volumes.needed=nvidia_driver com.nvidia.cuda.version=9.0 |
| ARG NCCL_VERSION |
| ARG CUBLAS_VERSION |
| ARG CUFFT_VERSION |
| ARG CURAND_VERSION |
| ARG CUSPARSE_VERSION |
| ARG CUSOLVER_VERSION |
| ARG CUTENSOR_VERSION |
| ARG NPP_VERSION |
| ARG NVJPEG_VERSION |
| ARG CUDNN_VERSION |
| ARG TRT_VERSION |
| ARG TRTOSS_VERSION |
| ARG NSIGHT_SYSTEMS_VERSION |
| ARG NSIGHT_COMPUTE_VERSION |
| ENV NCCL_VERSION=2.19.4 CUBLAS_VERSION=12.3.4.1 CUFFT_VERSION=11.0.12.1 CURAND_VERSION=10.3.4.107 CUSPARSE_VERSION=12.2.0.103 CUSOLVER_VERSION=11.5.4.101 CUTENSOR_VERSION=2.0.0.7 NPP_VERSION=12.2.3.2 NVJPEG_VERSION=12.3.0.81 CUDNN_VERSION=8.9.7.29+cuda12.2 TRT_VERSION=8.6.1.6+cuda12.0.1.011 TRTOSS_VERSION=23.11 NSIGHT_SYSTEMS_VERSION=2023.4.1.97 NSIGHT_COMPUTE_VERSION=2023.3.1.1 |
| RUN |17 CUDA_VERSION=12.3.2.001 CUDA_DRIVER_VERSION=545.23.08 JETPACK_HOST_MOUNTS= NCCL_VERSION=2.19.4 CUBLAS_VERSION=12.3.4.1 CUFFT_VERSION=11.0.12.1 CURAND_VERSION=10.3.4.107 CUSPARSE_VERSION=12.2.0.103 CUSOLVER_VERSION=11.5.4.101 CUTENSOR_VERSION=2.0.0.7 NPP_VERSION=12.2.3.2 NVJPEG_VERSION=12.3.0.81 CUDNN_VERSION=8.9.7.29+cuda12.2 TRT_VERSION=8.6.1.6+cuda12.0.1.011 TRTOSS_VERSION=23.11 NSIGHT_SYSTEMS_VERSION=2023.4.1.97 NSIGHT_COMPUTE_VERSION=2023.3.1.1 /bin/sh -c /nvidia/build-scripts/installNCCL.sh && /nvidia/build-scripts/installLIBS.sh && /nvidia/build-scripts/installCUDNN.sh && /nvidia/build-scripts/installTRT.sh && /nvidia/build-scripts/installNSYS.sh && /nvidia/build-scripts/installNCU.sh && /nvidia/build-scripts/installCUTENSOR.sh # buildkit |
| LABEL com.nvidia.nccl.version=2.19.4 com.nvidia.cublas.version=12.3.4.1 com.nvidia.cufft.version=11.0.12.1 com.nvidia.curand.version=10.3.4.107 com.nvidia.cusparse.version=12.2.0.103 com.nvidia.cusolver.version=11.5.4.101 com.nvidia.cutensor.version=2.0.0.7 com.nvidia.npp.version=12.2.3.2 com.nvidia.nvjpeg.version=12.3.0.81 com.nvidia.cudnn.version=8.9.7.29+cuda12.2 com.nvidia.tensorrt.version=8.6.1.6+cuda12.0.1.011 com.nvidia.tensorrtoss.version=23.11 com.nvidia.nsightsystems.version=2023.4.1.97 com.nvidia.nsightcompute.version=2023.3.1.1 |
| ARG DALI_VERSION |
| ARG DALI_BUILD |
| ARG POLYGRAPHY_VERSION |
| ARG TRANSFORMER_ENGINE_VERSION |
| ENV DALI_VERSION=1.33.0 DALI_BUILD=11414174 POLYGRAPHY_VERSION=0.49.1 TRANSFORMER_ENGINE_VERSION=1.2 |
| ADD docs.tgz / # buildkit |
| RUN |21 CUDA_VERSION=12.3.2.001 CUDA_DRIVER_VERSION=545.23.08 JETPACK_HOST_MOUNTS= NCCL_VERSION=2.19.4 CUBLAS_VERSION=12.3.4.1 CUFFT_VERSION=11.0.12.1 CURAND_VERSION=10.3.4.107 CUSPARSE_VERSION=12.2.0.103 CUSOLVER_VERSION=11.5.4.101 CUTENSOR_VERSION=2.0.0.7 NPP_VERSION=12.2.3.2 NVJPEG_VERSION=12.3.0.81 CUDNN_VERSION=8.9.7.29+cuda12.2 TRT_VERSION=8.6.1.6+cuda12.0.1.011 TRTOSS_VERSION=23.11 NSIGHT_SYSTEMS_VERSION=2023.4.1.97 NSIGHT_COMPUTE_VERSION=2023.3.1.1 DALI_VERSION=1.33.0 DALI_BUILD=11414174 POLYGRAPHY_VERSION=0.49.1 TRANSFORMER_ENGINE_VERSION=1.2 /bin/sh -c echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf && echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf # buildkit |
| ARG _LIBPATH_SUFFIX |
| ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin LD_LIBRARY_PATH=/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64 NVIDIA_VISIBLE_DEVICES=all NVIDIA_DRIVER_CAPABILITIES=compute,utility,video |
| COPY entrypoint/ /opt/nvidia/ # buildkit |
| ENV NVIDIA_PRODUCT_NAME=CUDA |
| ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"] |
| COPY NVIDIA_Deep_Learning_Container_License.pdf /workspace/ # buildkit |
| RUN /bin/sh -c export DEBIAN_FRONTEND=noninteractive && apt-get update && apt-get install -y --no-install-recommends build-essential git libglib2.0-0 less libnl-route-3-200 libnl-3-dev libnl-route-3-dev libnuma-dev libnuma1 libpmi2-0-dev nano numactl openssh-client vim wget && rm -rf /var/lib/apt/lists/* # buildkit |
| ARG GDRCOPY_VERSION |
| ARG HPCX_VERSION |
| ARG RDMACORE_VERSION |
| ARG MOFED_VERSION=5.4-rdmacore39.0 |
| ARG OPENUCX_VERSION |
| ARG OPENMPI_VERSION |
| ENV GDRCOPY_VERSION=2.3 HPCX_VERSION=2.16rc4 MOFED_VERSION=5.4-rdmacore39.0 OPENUCX_VERSION=1.15.0 OPENMPI_VERSION=4.1.5rc2 RDMACORE_VERSION=39.0 |
| ARG TARGETARCH |
| RUN |7 GDRCOPY_VERSION=2.3 HPCX_VERSION=2.16rc4 RDMACORE_VERSION=39.0 MOFED_VERSION=5.4-rdmacore39.0 OPENUCX_VERSION=1.15.0 OPENMPI_VERSION=4.1.5rc2 TARGETARCH=amd64 /bin/sh -c cd /nvidia && ( export DEBIAN_FRONTEND=noninteractive && apt-get update && apt-get install -y --no-install-recommends libibverbs1 libibverbs-dev librdmacm1 librdmacm-dev libibumad3 libibumad-dev ibverbs-utils ibverbs-providers && rm -rf /var/lib/apt/lists/* && rm $(dpkg-query -L libibverbs-dev librdmacm-dev libibumad-dev | grep "\(\.so\|\.a\)$") ) && ( cd opt/gdrcopy/ && dpkg -i libgdrapi_*.deb ) && ( cp -r opt/hpcx /opt/ && cp etc/ld.so.conf.d/hpcx.conf /etc/ld.so.conf.d/ && ln -sf /opt/hpcx/ompi /usr/local/mpi && ln -sf /opt/hpcx/ucx /usr/local/ucx && sed -i 's/^\(hwloc_base_binding_policy\) = core$/\1 = none/' /opt/hpcx/ompi/etc/openmpi-mca-params.conf && sed -i 's/^\(btl = self\)$/#\1/' /opt/hpcx/ompi/etc/openmpi-mca-params.conf ) && ldconfig # buildkit |
| ENV OPAL_PREFIX=/opt/hpcx/ompi PATH=/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin |
| ENV OMPI_MCA_coll_hcoll_enable=0 |
| COPY cuda-*.patch /tmp # buildkit |
| RUN |7 GDRCOPY_VERSION=2.3 HPCX_VERSION=2.16rc4 RDMACORE_VERSION=39.0 MOFED_VERSION=5.4-rdmacore39.0 OPENUCX_VERSION=1.15.0 OPENMPI_VERSION=4.1.5rc2 TARGETARCH=amd64 /bin/sh -c export DEVEL=1 BASE=0 && /nvidia/build-scripts/installNCU.sh && /nvidia/build-scripts/installCUDA.sh && /nvidia/build-scripts/installLIBS.sh && /nvidia/build-scripts/installNCCL.sh && /nvidia/build-scripts/installCUDNN.sh && /nvidia/build-scripts/installCUTENSOR.sh && /nvidia/build-scripts/installTRT.sh && /nvidia/build-scripts/installNSYS.sh && if [ -f "/tmp/cuda-${_CUDA_VERSION_MAJMIN}.patch" ]; then patch -p0 < /tmp/cuda-${_CUDA_VERSION_MAJMIN}.patch; fi && rm -f /tmp/cuda-*.patch # buildkit |
| ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs: |
| ENV NVIDIA_PRODUCT_NAME=NVIDIA TensorRT |
| ENV TENSORRT_VERSION=8.6.1.6+cuda12.0.1.011 |
| LABEL com.nvidia.tensorrt.version=8.6.1.6+cuda12.0.1.011 |
| ARG NVIDIA_TENSORRT_VERSION |
| ENV NVIDIA_TENSORRT_VERSION=24.01 |
| ARG TRTOSS_VERSION=master |
| ENV TENSORRT_OSS_VERSION=master |
| ARG PYVER=2.7 |
| RUN |3 NVIDIA_TENSORRT_VERSION=24.01 TRTOSS_VERSION=master PYVER=3.10 /bin/sh -c if [ "$PYVER" == "3.8" ]; then DISTUTILS="python3-distutils"; fi # buildkit |
| RUN |3 NVIDIA_TENSORRT_VERSION=24.01 TRTOSS_VERSION=master PYVER=3.10 /bin/sh -c PYSFX=`[ "$PYVER" != "2.7" ] && echo "$PYVER" | cut -c1-1 || echo ""` && apt-get update && DEBIAN_FRONTEND="noninteractive" apt-get install -y --no-install-recommends tzdata && apt-get install -y --no-install-recommends python$PYSFX python$PYVER-dev $DISTUTILS python$PYSFX-pip libssl-dev unzip && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN |3 NVIDIA_TENSORRT_VERSION=24.01 TRTOSS_VERSION=master PYVER=3.10 /bin/sh -c CMAKE_VERSION=3.24 CMAKE_PATCH=0 CMAKE_DIR=cmake-${CMAKE_VERSION}.${CMAKE_PATCH} && cd /usr/local/src && wget https://cmake.org/files/v${CMAKE_VERSION}/${CMAKE_DIR}.tar.gz && tar -xvf ${CMAKE_DIR}.tar.gz && cd ${CMAKE_DIR} && ./bootstrap --parallel=$(nproc) && make -j$(nproc) install && cd .. && rm -rf ${CMAKE_DIR}* # buildkit |
| RUN |3 NVIDIA_TENSORRT_VERSION=24.01 TRTOSS_VERSION=master PYVER=3.10 /bin/sh -c cd ${HOME} && git clone -b v2.2.3 https://github.com/pybind/pybind11.git # buildkit |
| RUN |3 NVIDIA_TENSORRT_VERSION=24.01 TRTOSS_VERSION=master PYVER=3.10 /bin/sh -c rm -f /usr/bin/python && ln -s /usr/bin/python$PYVER /usr/bin/python # buildkit |
| RUN |3 NVIDIA_TENSORRT_VERSION=24.01 TRTOSS_VERSION=master PYVER=3.10 /bin/sh -c MAJ=`echo "$PYVER" | cut -c1-1` && rm -f /usr/bin/python$MAJ && ln -s /usr/bin/python$PYVER /usr/bin/python$MAJ # buildkit |
| RUN |3 NVIDIA_TENSORRT_VERSION=24.01 TRTOSS_VERSION=master PYVER=3.10 /bin/sh -c curl -O https://bootstrap.pypa.io/get-pip.py && python get-pip.py && rm get-pip.py # buildkit |
| RUN |3 NVIDIA_TENSORRT_VERSION=24.01 TRTOSS_VERSION=master PYVER=3.10 /bin/sh -c pip install --upgrade --no-cache-dir 'numpy<1.24' && pip install --upgrade --no-cache-dir pycuda pillow # buildkit |
| RUN |3 NVIDIA_TENSORRT_VERSION=24.01 TRTOSS_VERSION=master PYVER=3.10 /bin/sh -c rm -f /etc/dpkg/dpkg.cfg.d/excludes # buildkit |
| RUN |3 NVIDIA_TENSORRT_VERSION=24.01 TRTOSS_VERSION=master PYVER=3.10 /bin/sh -c URL=$(VERIFY=1 BASE=0 DEVEL=0 SAMPLES=0 PYTHON=0 /nvidia/build-scripts/installTRT.sh 2>/dev/null | sed -n "s/^.*\(http.*\)tar.*$/\1/p")tar && FILE=$(wget -O - $URL 2>/dev/null | sed -n 's/^.*href="\(TensorRT[^"]*\)".*$/\1/p' | egrep -v "internal|safety") && wget --quiet $URL/$FILE -O - | tar -xz && PY=$(python -c 'import sys; print(str(sys.version_info[0])+str(sys.version_info[1]))') && pip install TensorRT-*/python/tensorrt-*-cp$PY*.whl && pip install TensorRT-*/graphsurgeon/graphsurgeon-*.whl && pip install TensorRT-*/uff/uff-*.whl && mv /usr/src/tensorrt /opt/ && mv TensorRT-*/samples /opt/tensorrt/ && mv TensorRT-*/data /opt/tensorrt/ && ln -s /opt/tensorrt /usr/src/tensorrt && rm -r TensorRT-* && UFF_PATH=$(pip show uff | sed -n 's/Location: \(.*\)$/\1/p')/uff && sed -i 's/from tensorflow import GraphDef/from tensorflow.python import GraphDef/' $UFF_PATH/converters/tensorflow/conversion_helpers.py && chmod +x ${UFF_PATH}/bin/convert_to_uff.py && ln -sf ${UFF_PATH}/bin/convert_to_uff.py /usr/local/bin/convert-to-uff # buildkit |
| RUN |3 NVIDIA_TENSORRT_VERSION=24.01 TRTOSS_VERSION=master PYVER=3.10 /bin/sh -c pip install --no-cache-dir --extra-index-url https://pypi.ngc.nvidia.com --extra-index-url https://tensorrt-read-only:Tensorrt\@123@urm.nvidia.com/artifactory/api/pypi/sw-tensorrt-pypi/simple polygraphy # buildkit |
| COPY python_setup.sh /opt/tensorrt/python/python_setup.sh # buildkit |
| COPY install_opensource.sh /opt/tensorrt/ # buildkit |
| ENV PATH=/opt/tensorrt/bin:/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin |
| COPY trt8.5-samples-cuda12.patch /tmp/ # buildkit |
| RUN |3 NVIDIA_TENSORRT_VERSION=24.01 TRTOSS_VERSION=master PYVER=3.10 /bin/sh -c cd /opt/tensorrt/samples && patch -p2 < /tmp/trt8.5-samples-cuda12.patch && rm /tmp/trt8.5-samples-cuda12.patch # buildkit |
| RUN |3 NVIDIA_TENSORRT_VERSION=24.01 TRTOSS_VERSION=master PYVER=3.10 /bin/sh -c cd /opt/tensorrt/samples/python/introductory_parser_samples && sed -i 's/Image\.ANTIALIAS/Image.LANCZOS/g' onnx_resnet50.py # buildkit |
| RUN |3 NVIDIA_TENSORRT_VERSION=24.01 TRTOSS_VERSION=master PYVER=3.10 /bin/sh -c mkdir -p /workspace/tensorrt && cp -r /opt/tensorrt/bin /workspace/tensorrt && cp -r /opt/tensorrt/samples /workspace/tensorrt && ln -s /opt/tensorrt/data /workspace/tensorrt && chmod -R a+w /workspace/tensorrt/bin && chmod -R a+w /workspace/tensorrt/samples # buildkit |
| ADD docs.tgz / # buildkit |
| WORKDIR /workspace |
| COPY README.md README.md # buildkit |
| RUN |3 NVIDIA_TENSORRT_VERSION=24.01 TRTOSS_VERSION=master PYVER=3.10 /bin/sh -c ln -sf ${_CUDA_COMPAT_PATH}/lib.real ${_CUDA_COMPAT_PATH}/lib && echo ${_CUDA_COMPAT_PATH}/lib > /etc/ld.so.conf.d/00-cuda-compat.conf && ldconfig && rm -f ${_CUDA_COMPAT_PATH}/lib # buildkit |
| COPY entrypoint.d/ /opt/nvidia/entrypoint.d/ # buildkit |
| ARG NVIDIA_BUILD_ID |
| ENV NVIDIA_BUILD_ID=78767117 |
| LABEL com.nvidia.build.id=78767117 |
| ARG NVIDIA_BUILD_REF |
| LABEL com.nvidia.build.ref=dcbd5e3b6f52055788b46a9c09a5212040896748 |
| WORKDIR /app |
| COPY . /app # buildkit |
| RUN /bin/sh -c pip install --no-cache-dir -r requirements.txt # buildkit |
| RUN /bin/sh -c pip3 install torch --index-url https://download.pytorch.org/whl/cu118 # buildkit |
