chore: import upstream snapshot with attribution
Docker Image CI / build-ubuntu2004 (push) Waiting to run
Docker Image CI / build-ubuntu2004 (push) Waiting to run
This commit is contained in:
wehub-resource-sync
Executable
+50
@@ -0,0 +1,50 @@
|
||||
#!/usr/bin/env bash
|
||||
#
|
||||
# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
#
|
||||
|
||||
arg_dockerfile=docker/ubuntu-24.04.Dockerfile
|
||||
arg_imagename=tensorrt-ubuntu
|
||||
arg_help=0
|
||||
|
||||
while [[ "$#" -gt 0 ]]; do case $1 in
|
||||
--file) arg_dockerfile="$2"; shift;;
|
||||
--tag) arg_imagename="$2"; shift;;
|
||||
--cuda) arg_cudaversion="$2"; shift;;
|
||||
-h|--help) arg_help=1;;
|
||||
*) echo "Unknown parameter passed: $1"; echo "For help type: $0 --help"; exit 1;
|
||||
esac; shift; done
|
||||
|
||||
if [ "$arg_help" -eq "1" ]; then
|
||||
echo "Usage: $0 [options]"
|
||||
echo " --help or -h : Print this help menu."
|
||||
echo " --file <dockerfile> : Docker file to use for build."
|
||||
echo " --tag <imagename> : Image name for the generated container."
|
||||
echo " --cuda <version> : CUDA version to use."
|
||||
exit;
|
||||
fi
|
||||
|
||||
if [ -z "$arg_cudaversion" ]
|
||||
then
|
||||
echo "--cuda not specified, so not passing in --build-arg CUDA_VERSION to Dockerfile"
|
||||
docker_args="-f $arg_dockerfile --build-arg uid=$(id -u) --build-arg gid=$(id -g) --tag=$arg_imagename ."
|
||||
else
|
||||
docker_args="-f $arg_dockerfile --build-arg CUDA_VERSION=$arg_cudaversion --build-arg CUDA_VERSION_MAJOR_MINOR=${arg_cudaversion:0:4} --build-arg uid=$(id -u) --build-arg gid=$(id -g) --tag=$arg_imagename ."
|
||||
fi
|
||||
|
||||
echo "Building container:"
|
||||
echo "> docker build $docker_args"
|
||||
docker build $docker_args
|
||||
Executable
+58
@@ -0,0 +1,58 @@
|
||||
#!/usr/bin/env bash
|
||||
#
|
||||
# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
#
|
||||
|
||||
arg_tag=tensorrt-ubuntu24.04
|
||||
arg_gpus=all
|
||||
arg_jupyter=0
|
||||
arg_help=0
|
||||
|
||||
while [[ "$#" -gt 0 ]]; do case $1 in
|
||||
--tag) arg_tag="$2"; shift;;
|
||||
--gpus) arg_gpus="$2"; shift;;
|
||||
--jupyter) arg_jupyter="$2"; shift;;
|
||||
-h|--help) arg_help=1;;
|
||||
*) echo "Unknown parameter passed: $1"; echo "For help type: $0 --help"; exit 1;
|
||||
esac; shift; done
|
||||
|
||||
if [ "$arg_help" -eq "1" ]; then
|
||||
echo "Usage: $0 [options]"
|
||||
echo " --help or -h : Print this help menu."
|
||||
echo " --tag <imagetag> : Image name for generated container."
|
||||
echo " --gpus <number> : Number of GPUs visible in container. Set 'none' to disable, and 'all' to make all visible."
|
||||
echo " --jupyter <port> : Launch Jupyter notebook using the specified port number."
|
||||
exit;
|
||||
fi
|
||||
|
||||
extra_args=""
|
||||
if [ "$arg_gpus" != "none" ]; then
|
||||
extra_args="$extra_args --gpus $arg_gpus"
|
||||
fi
|
||||
|
||||
if [ "$arg_jupyter" -ne "0" ]; then
|
||||
extra_args+=" -p $arg_jupyter:$arg_jupyter"
|
||||
fi
|
||||
|
||||
docker_args="$extra_args -v ${PWD}:/workspace/TensorRT --rm -it $arg_tag:latest"
|
||||
|
||||
if [ "$arg_jupyter" -ne "0" ]; then
|
||||
docker_args+=" jupyter-lab --port=$arg_jupyter --no-browser --ip 0.0.0.0 --allow-root"
|
||||
fi
|
||||
|
||||
echo "Launching container:"
|
||||
echo "> docker run $docker_args"
|
||||
docker run $docker_args
|
||||
@@ -0,0 +1,112 @@
|
||||
#
|
||||
# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
#
|
||||
|
||||
ARG CUDA_VERSION=13.3.0
|
||||
|
||||
FROM nvidia/cuda:${CUDA_VERSION}-devel-rockylinux8
|
||||
LABEL maintainer="NVIDIA CORPORATION"
|
||||
|
||||
ENV TRT_VERSION 11.1.0.106
|
||||
SHELL ["/bin/bash", "-c"]
|
||||
|
||||
# Setup user account
|
||||
ARG uid=1000
|
||||
ARG gid=1000
|
||||
RUN groupadd -r -f -g ${gid} trtuser && useradd -o -r -l -u ${uid} -g ${gid} -ms /bin/bash trtuser
|
||||
RUN usermod -aG wheel trtuser
|
||||
RUN echo 'trtuser:nvidia' | chpasswd
|
||||
RUN mkdir -p /workspace && chown trtuser /workspace
|
||||
|
||||
# Install requried packages
|
||||
RUN dnf -y groupinstall "Development Tools"
|
||||
RUN dnf -y install \
|
||||
openssl-devel \
|
||||
bzip2-devel \
|
||||
libffi-devel \
|
||||
wget \
|
||||
perl-core \
|
||||
git \
|
||||
pkg-config \
|
||||
unzip \
|
||||
sudo \
|
||||
libnccl \
|
||||
libnccl-devel \
|
||||
openmpi \
|
||||
openmpi-devel \
|
||||
zstd \
|
||||
epel-release
|
||||
|
||||
RUN dnf -y install ccache
|
||||
|
||||
# Install python3
|
||||
RUN dnf install -y python38 python38-devel &&\
|
||||
cd /usr/bin && ln -s /usr/bin/pip3.8 pip;
|
||||
|
||||
|
||||
# Install TensorRT
|
||||
RUN if [ "${CUDA_VERSION:0:2}" = "13" ]; then \
|
||||
wget https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/11.1.0/tars/TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-13.3-Release-external.tar.zst \
|
||||
&& tar --use-compress-program=unzstd -xf TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-13.3-Release-external.tar.zst \
|
||||
&& cp -a TensorRT-11.1.0.106/lib/*.so* /usr/lib64 \
|
||||
&& pip install TensorRT-11.1.0.106/python/tensorrt-11.1.0.106-cp38-none-linux_x86_64.whl ;\
|
||||
elif [ "${CUDA_VERSION:0:2}" = "12" ]; then \
|
||||
wget https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/11.1.0/tars/TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-12.9-Release-external.tar.zst \
|
||||
&& tar --use-compress-program=unzstd -xf TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-12.9-Release-external.tar.zst \
|
||||
&& cp -a TensorRT-11.1.0.106/lib/*.so* /usr/lib64 \
|
||||
&& pip install TensorRT-11.1.0.106/python/tensorrt-11.1.0.106-cp38-none-linux_x86_64.whl ;\
|
||||
else \
|
||||
echo "Invalid CUDA_VERSION"; \
|
||||
exit 1; \
|
||||
fi
|
||||
|
||||
# Install PyPI packages
|
||||
RUN pip install --upgrade pip
|
||||
RUN pip install setuptools>=41.0.0
|
||||
RUN pip install numpy
|
||||
RUN pip install jupyter jupyterlab
|
||||
|
||||
# Install Cmake
|
||||
RUN cd /tmp && \
|
||||
wget https://github.com/Kitware/CMake/releases/download/v3.31.11/cmake-3.31.11-Linux-x86_64.sh && \
|
||||
chmod +x cmake-3.31.11-Linux-x86_64.sh && \
|
||||
./cmake-3.31.11-Linux-x86_64.sh --prefix=/usr/local --exclude-subdir --skip-license && \
|
||||
rm ./cmake-3.31.11-Linux-x86_64.sh
|
||||
|
||||
# Install gtest
|
||||
RUN cd /tmp && \
|
||||
git clone https://github.com/google/googletest.git -b v1.14.0 && \
|
||||
cd googletest && \
|
||||
mkdir build && cd build && \
|
||||
cmake .. && \
|
||||
make -j4 && \
|
||||
make install && \
|
||||
rm -rf /tmp/googletest
|
||||
|
||||
# Download NGC client
|
||||
RUN cd /usr/local/bin && wget https://ngc.nvidia.com/downloads/ngccli_cat_linux.zip && unzip ngccli_cat_linux.zip && chmod u+x ngc-cli/ngc && rm ngccli_cat_linux.zip ngc-cli.md5 && echo "no-apikey\nascii\n" | ngc-cli/ngc config set
|
||||
|
||||
RUN ln -s /usr/bin/python3 /usr/bin/python
|
||||
|
||||
# Set environment and working directory
|
||||
ENV TRT_LIBPATH /usr/lib64
|
||||
ENV TRT_OSSPATH /workspace/TensorRT
|
||||
ENV PATH="/workspace/TensorRT/build/out:${PATH}:/usr/local/bin/ngc-cli"
|
||||
ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:${TRT_OSSPATH}/build/out:${TRT_LIBPATH}"
|
||||
WORKDIR /workspace
|
||||
|
||||
USER trtuser
|
||||
RUN ["/bin/bash"]
|
||||
@@ -0,0 +1,112 @@
|
||||
#
|
||||
# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
#
|
||||
|
||||
ARG CUDA_VERSION=13.3.0
|
||||
|
||||
FROM nvidia/cuda:${CUDA_VERSION}-devel-rockylinux9
|
||||
LABEL maintainer="NVIDIA CORPORATION"
|
||||
|
||||
ENV TRT_VERSION 11.1.0.106
|
||||
SHELL ["/bin/bash", "-c"]
|
||||
|
||||
# Setup user account
|
||||
ARG uid=1000
|
||||
ARG gid=1000
|
||||
RUN groupadd -r -f -g ${gid} trtuser && useradd -o -r -l -u ${uid} -g ${gid} -ms /bin/bash trtuser
|
||||
RUN usermod -aG wheel trtuser
|
||||
RUN echo 'trtuser:nvidia' | chpasswd
|
||||
RUN mkdir -p /workspace && chown trtuser /workspace
|
||||
|
||||
# Install python3
|
||||
RUN dnf install -y python39 python3-devel && \
|
||||
cd /usr/bin && rm pip && ln -s /usr/bin/pip3.9 pip;
|
||||
|
||||
# Install PyPI packages
|
||||
RUN pip install --upgrade pip
|
||||
RUN pip install setuptools>=41.0.0
|
||||
RUN pip install numpy
|
||||
RUN pip install jupyter jupyterlab
|
||||
|
||||
# Install requried packages
|
||||
RUN dnf -y groupinstall "Development Tools"
|
||||
RUN dnf -y install \
|
||||
openssl-devel \
|
||||
bzip2-devel \
|
||||
libffi-devel \
|
||||
wget \
|
||||
perl-core \
|
||||
git \
|
||||
pkg-config \
|
||||
unzip \
|
||||
sudo \
|
||||
libnccl \
|
||||
libnccl-devel \
|
||||
openmpi \
|
||||
openmpi-devel \
|
||||
zstd \
|
||||
epel-release
|
||||
|
||||
RUN dnf -y install ccache
|
||||
|
||||
# Install TensorRT
|
||||
RUN if [ "${CUDA_VERSION:0:2}" = "13" ]; then \
|
||||
wget https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/11.1.0/tars/TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-13.3-Release-external.tar.zst \
|
||||
&& tar --use-compress-program=unzstd -xf TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-13.3-Release-external.tar.zst \
|
||||
&& cp -a TensorRT-11.1.0.106/lib/*.so* /usr/lib64 \
|
||||
&& pip install TensorRT-11.1.0.106/python/tensorrt-11.1.0.106-cp39-none-linux_x86_64.whl ;\
|
||||
elif [ "${CUDA_VERSION:0:2}" = "12" ]; then \
|
||||
wget https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/11.1.0/tars/TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-12.9-Release-external.tar.zst \
|
||||
&& tar --use-compress-program=unzstd -xf TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-12.9-Release-external.tar.zst \
|
||||
&& cp -a TensorRT-11.1.0.106/lib/*.so* /usr/lib64 \
|
||||
&& pip install TensorRT-11.1.0.106/python/tensorrt-11.1.0.106-cp39-none-linux_x86_64.whl ;\
|
||||
else \
|
||||
echo "Invalid CUDA_VERSION"; \
|
||||
exit 1; \
|
||||
fi
|
||||
|
||||
|
||||
# Install Cmake
|
||||
RUN cd /tmp && \
|
||||
wget https://github.com/Kitware/CMake/releases/download/v3.31.11/cmake-3.31.11-Linux-x86_64.sh && \
|
||||
chmod +x cmake-3.31.11-Linux-x86_64.sh && \
|
||||
./cmake-3.31.11-Linux-x86_64.sh --prefix=/usr/local --exclude-subdir --skip-license && \
|
||||
rm ./cmake-3.31.11-Linux-x86_64.sh
|
||||
|
||||
# Install gtest
|
||||
RUN cd /tmp && \
|
||||
git clone https://github.com/google/googletest.git -b v1.14.0 && \
|
||||
cd googletest && \
|
||||
mkdir build && cd build && \
|
||||
cmake .. && \
|
||||
make -j4 && \
|
||||
make install && \
|
||||
rm -rf /tmp/googletest
|
||||
|
||||
# Download NGC client
|
||||
RUN cd /usr/local/bin && wget https://ngc.nvidia.com/downloads/ngccli_cat_linux.zip && unzip ngccli_cat_linux.zip && chmod u+x ngc-cli/ngc && rm ngccli_cat_linux.zip ngc-cli.md5 && echo "no-apikey\nascii\n" | ngc-cli/ngc config set
|
||||
|
||||
RUN ln -s /usr/bin/python3 /usr/bin/python
|
||||
|
||||
# Set environment and working directory
|
||||
ENV TRT_LIBPATH /usr/lib64
|
||||
ENV TRT_OSSPATH /workspace/TensorRT
|
||||
ENV PATH="/workspace/TensorRT/build/out:${PATH}:/usr/local/bin/ngc-cli"
|
||||
ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:${TRT_OSSPATH}/build/out:${TRT_LIBPATH}"
|
||||
WORKDIR /workspace
|
||||
|
||||
USER trtuser
|
||||
RUN ["/bin/bash"]
|
||||
@@ -0,0 +1,128 @@
|
||||
#
|
||||
# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
#
|
||||
|
||||
ARG CUDA_VERSION=13.3.0
|
||||
|
||||
FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04
|
||||
LABEL maintainer="NVIDIA CORPORATION"
|
||||
|
||||
ENV TRT_VERSION 11.1.0.106
|
||||
SHELL ["/bin/bash", "-c"]
|
||||
|
||||
# Setup user account
|
||||
ARG uid=1000
|
||||
ARG gid=1000
|
||||
RUN groupadd -r -f -g ${gid} trtuser && useradd -o -r -l -u ${uid} -g ${gid} -ms /bin/bash trtuser
|
||||
RUN usermod -aG sudo trtuser
|
||||
RUN echo 'trtuser:nvidia' | chpasswd
|
||||
RUN mkdir -p /workspace && chown trtuser /workspace
|
||||
|
||||
# Required to build Ubuntu 22.04 without user prompts with DLFW container
|
||||
ENV DEBIAN_FRONTEND=noninteractive
|
||||
|
||||
# Update CUDA signing key
|
||||
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub
|
||||
|
||||
# Install requried libraries
|
||||
RUN apt-get update && apt-get install -y software-properties-common
|
||||
RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||
libcurl4-openssl-dev \
|
||||
wget \
|
||||
git \
|
||||
pkg-config \
|
||||
sudo \
|
||||
ssh \
|
||||
libssl-dev \
|
||||
pbzip2 \
|
||||
pv \
|
||||
bzip2 \
|
||||
unzip \
|
||||
devscripts \
|
||||
lintian \
|
||||
fakeroot \
|
||||
dh-make \
|
||||
build-essential \
|
||||
libnccl2 \
|
||||
libnccl-dev \
|
||||
openmpi-bin \
|
||||
libopenmpi-dev \
|
||||
zstd \
|
||||
ccache
|
||||
|
||||
# Install python3
|
||||
RUN apt-get install -y --no-install-recommends \
|
||||
python3 \
|
||||
python3-pip \
|
||||
python3-dev \
|
||||
python3-wheel &&\
|
||||
cd /usr/local/bin &&\
|
||||
ln -s /usr/bin/python3 python &&\
|
||||
ln -s /usr/bin/pip3 pip;
|
||||
|
||||
# Install TensorRT
|
||||
RUN if [ "${CUDA_VERSION:0:2}" = "13" ]; then \
|
||||
wget https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/11.1.0/tars/TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-13.3-Release-external.tar.zst \
|
||||
&& tar -xf TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-13.3-Release-external.tar.zst \
|
||||
&& cp -a TensorRT-11.1.0.106/lib/*.so* /usr/lib/x86_64-linux-gnu \
|
||||
&& pip install TensorRT-11.1.0.106/python/tensorrt-11.1.0.106-cp310-none-linux_x86_64.whl ;\
|
||||
elif [ "${CUDA_VERSION:0:2}" = "12" ]; then \
|
||||
wget https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/11.1.0/tars/TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-12.9-Release-external.tar.zst \
|
||||
&& tar -xf TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-12.9-Release-external.tar.zst \
|
||||
&& cp -a TensorRT-11.1.0.106/lib/*.so* /usr/lib/x86_64-linux-gnu \
|
||||
&& pip install TensorRT-11.1.0.106/python/tensorrt-11.1.0.106-cp310-none-linux_x86_64.whl ;\
|
||||
else \
|
||||
echo "Invalid CUDA_VERSION"; \
|
||||
exit 1; \
|
||||
fi
|
||||
|
||||
|
||||
# Install PyPI packages
|
||||
RUN pip3 install --upgrade pip
|
||||
RUN pip3 install setuptools>=41.0.0
|
||||
RUN pip3 install jupyter jupyterlab
|
||||
# Workaround to remove numpy installed with tensorflow
|
||||
RUN pip3 install --upgrade numpy
|
||||
|
||||
# Install Cmake
|
||||
RUN cd /tmp && \
|
||||
wget https://github.com/Kitware/CMake/releases/download/v3.31.11/cmake-3.31.11-Linux-x86_64.sh && \
|
||||
chmod +x cmake-3.31.11-Linux-x86_64.sh && \
|
||||
./cmake-3.31.11-Linux-x86_64.sh --prefix=/usr/local --exclude-subdir --skip-license && \
|
||||
rm ./cmake-3.31.11-Linux-x86_64.sh
|
||||
|
||||
# Install gtest
|
||||
RUN cd /tmp && \
|
||||
git clone https://github.com/google/googletest.git -b v1.14.0 && \
|
||||
cd googletest && \
|
||||
mkdir build && cd build && \
|
||||
cmake .. && \
|
||||
make -j4 && \
|
||||
make install && \
|
||||
rm -rf /tmp/googletest
|
||||
|
||||
# Download NGC client
|
||||
RUN cd /usr/local/bin && wget https://ngc.nvidia.com/downloads/ngccli_cat_linux.zip && unzip ngccli_cat_linux.zip && chmod u+x ngc-cli/ngc && rm ngccli_cat_linux.zip ngc-cli.md5 && echo "no-apikey\nascii\n" | ngc-cli/ngc config set
|
||||
|
||||
# Set environment and working directory
|
||||
ENV TRT_LIBPATH /usr/lib/x86_64-linux-gnu
|
||||
ENV TRT_OSSPATH /workspace/TensorRT
|
||||
ENV PATH="/workspace/TensorRT/build/out:${PATH}:/usr/local/bin/ngc-cli"
|
||||
ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:${TRT_OSSPATH}/build/out:${TRT_LIBPATH}"
|
||||
WORKDIR /workspace
|
||||
|
||||
USER trtuser
|
||||
RUN ["/bin/bash"]
|
||||
@@ -0,0 +1,137 @@
|
||||
#
|
||||
# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
#
|
||||
|
||||
ARG CUDA_VERSION=13.3.0
|
||||
|
||||
# Multi-arch container support available in non-cudnn containers.
|
||||
FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu24.04
|
||||
|
||||
ENV TRT_VERSION 11.1.0.106
|
||||
SHELL ["/bin/bash", "-c"]
|
||||
|
||||
# Setup user account and edit default account
|
||||
RUN if id "ubuntu" &>/dev/null; then \
|
||||
usermod -u 1234 ubuntu && \
|
||||
groupmod -g 1234 ubuntu; \
|
||||
fi
|
||||
ARG uid=1000
|
||||
ARG gid=1000
|
||||
RUN groupadd -r -f -g ${gid} trtuser && useradd -o -r -l -u ${uid} -g ${gid} -ms /bin/bash trtuser
|
||||
RUN usermod -aG sudo trtuser
|
||||
RUN echo 'trtuser:nvidia' | chpasswd
|
||||
RUN mkdir -p /workspace && chown trtuser /workspace
|
||||
|
||||
# Required to build Ubuntu 20.04 without user prompts with DLFW container
|
||||
ENV DEBIAN_FRONTEND=noninteractive
|
||||
|
||||
# Update CUDA signing key
|
||||
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/sbsa/3bf863cc.pub
|
||||
|
||||
# Install requried libraries
|
||||
RUN apt-get update && apt-get install -y software-properties-common
|
||||
RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||
libcurl4-openssl-dev \
|
||||
wget \
|
||||
git \
|
||||
pkg-config \
|
||||
sudo \
|
||||
ssh \
|
||||
libssl-dev \
|
||||
pbzip2 \
|
||||
pv \
|
||||
bzip2 \
|
||||
unzip \
|
||||
devscripts \
|
||||
lintian \
|
||||
fakeroot \
|
||||
dh-make \
|
||||
build-essential \
|
||||
libffi-dev \
|
||||
libnccl2 \
|
||||
libnccl-dev \
|
||||
openmpi-bin \
|
||||
libopenmpi-dev \
|
||||
zstd \
|
||||
ccache
|
||||
|
||||
# Install python3
|
||||
RUN apt-get install -y --no-install-recommends \
|
||||
python3 \
|
||||
python3-pip \
|
||||
python3-dev \
|
||||
python3-wheel \
|
||||
python3-venv &&\
|
||||
cd /usr/local/bin &&\
|
||||
ln -s /usr/bin/python3 python &&\
|
||||
ln -s /usr/bin/pip3 pip;
|
||||
|
||||
# Create python3 virtualenv
|
||||
RUN python3 -m venv /opt/venv
|
||||
ENV PATH="/opt/venv/bin:$PATH"
|
||||
|
||||
# Install TensorRT
|
||||
RUN if [ "${CUDA_VERSION:0:2}" = "13" ]; then \
|
||||
wget https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/11.1.0/tars/TensorRT-Enterprise-11.1.0.106-Linux-aarch64-cuda-13.3-Release-external.tar.zst \
|
||||
&& tar -xf TensorRT-Enterprise-11.1.0.106-Linux-aarch64-cuda-13.3-Release-external.tar.zst \
|
||||
&& cp -a TensorRT-11.1.0.106/lib/*.so* /usr/lib/aarch64-linux-gnu/ \
|
||||
&& pip install TensorRT-11.1.0.106/python/tensorrt-11.1.0.106-cp312-none-linux_aarch64.whl ;\
|
||||
elif [ "${CUDA_VERSION:0:2}" = "12" ]; then \
|
||||
wget https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/11.1.0/tars/TensorRT-Enterprise-11.1.0.106-Linux-aarch64-cuda-12.9-Release-external.tar.zst \
|
||||
&& tar -xf TensorRT-Enterprise-11.1.0.106-Linux-aarch64-cuda-12.9-Release-external.tar.zst \
|
||||
&& cp -a TensorRT-11.1.0.106/lib/*.so* /usr/lib/aarch64-linux-gnu/ \
|
||||
&& pip install TensorRT-11.1.0.106/python/tensorrt-11.1.0.106-cp312-none-linux_aarch64.whl ;\
|
||||
else \
|
||||
echo "Invalid CUDA_VERSION"; \
|
||||
exit 1; \
|
||||
fi
|
||||
|
||||
# Install Cmake
|
||||
RUN cd /tmp && \
|
||||
wget https://github.com/Kitware/CMake/releases/download/v3.31.11/cmake-3.31.11-linux-aarch64.sh && \
|
||||
chmod +x cmake-3.31.11-linux-aarch64.sh && \
|
||||
./cmake-3.31.11-linux-aarch64.sh --prefix=/usr/local --exclude-subdir --skip-license && \
|
||||
rm ./cmake-3.31.11-linux-aarch64.sh
|
||||
|
||||
# Install gtest
|
||||
RUN cd /tmp && \
|
||||
git clone https://github.com/google/googletest.git -b v1.14.0 && \
|
||||
cd googletest && \
|
||||
mkdir build && cd build && \
|
||||
cmake .. && \
|
||||
make -j4 && \
|
||||
make install && \
|
||||
rm -rf /tmp/googletest
|
||||
|
||||
# Install PyPI packages
|
||||
RUN pip3 install --upgrade pip
|
||||
RUN pip3 install setuptools>=41.0.0
|
||||
RUN pip3 install jupyter jupyterlab
|
||||
# Workaround to remove numpy installed with tensorflow
|
||||
RUN pip3 install --upgrade numpy
|
||||
|
||||
# Download NGC client
|
||||
RUN cd /usr/local/bin && wget https://ngc.nvidia.com/downloads/ngccli_arm64.zip && unzip ngccli_arm64.zip && chmod u+x ngc-cli/ngc && rm ngccli_arm64.zip ngc-cli.md5 && echo "no-apikey\nascii\n" | ngc-cli/ngc config set
|
||||
|
||||
# Set environment and working directory
|
||||
ENV TRT_LIBPATH /usr/lib/aarch64-linux-gnu/
|
||||
ENV TRT_OSSPATH /workspace/TensorRT
|
||||
ENV PATH="/workspace/TensorRT/build/out:${PATH}:/usr/local/bin/ngc-cli"
|
||||
ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:${TRT_OSSPATH}/build/out:${TRT_LIBPATH}"
|
||||
WORKDIR /workspace
|
||||
|
||||
USER trtuser
|
||||
RUN ["/bin/bash"]
|
||||
@@ -0,0 +1,136 @@
|
||||
#
|
||||
# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
#
|
||||
|
||||
ARG CUDA_VERSION=13.3.0
|
||||
|
||||
# Multi-arch container support available in non-cudnn containers.
|
||||
FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu24.04
|
||||
LABEL maintainer="NVIDIA CORPORATION"
|
||||
|
||||
ENV TRT_VERSION=11.1.0.106
|
||||
SHELL ["/bin/bash", "-c"]
|
||||
|
||||
# Setup user account and edit default account
|
||||
RUN if id "ubuntu" &>/dev/null; then \
|
||||
usermod -u 1234 ubuntu && \
|
||||
groupmod -g 1234 ubuntu; \
|
||||
fi
|
||||
ARG uid=1000
|
||||
ARG gid=1000
|
||||
RUN groupadd -r -f -g ${gid} trtuser && useradd -o -r -l -u ${uid} -g ${gid} -ms /bin/bash trtuser
|
||||
RUN usermod -aG sudo trtuser
|
||||
RUN echo 'trtuser:nvidia' | chpasswd
|
||||
RUN mkdir -p /workspace && chown trtuser /workspace
|
||||
|
||||
ENV DEBIAN_FRONTEND=noninteractive
|
||||
|
||||
# Update CUDA signing key
|
||||
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/3bf863cc.pub
|
||||
|
||||
# Install requried libraries
|
||||
RUN apt-get update && apt-get install -y software-properties-common
|
||||
RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||
libcurl4-openssl-dev \
|
||||
wget \
|
||||
git \
|
||||
pkg-config \
|
||||
sudo \
|
||||
ssh \
|
||||
libssl-dev \
|
||||
pbzip2 \
|
||||
pv \
|
||||
bzip2 \
|
||||
unzip \
|
||||
devscripts \
|
||||
lintian \
|
||||
fakeroot \
|
||||
dh-make \
|
||||
build-essential \
|
||||
libnccl2 \
|
||||
libnccl-dev \
|
||||
openmpi-bin \
|
||||
libopenmpi-dev \
|
||||
zstd \
|
||||
ccache
|
||||
|
||||
# Install python3
|
||||
RUN apt-get install -y --no-install-recommends \
|
||||
python3 \
|
||||
python3-pip \
|
||||
python3-dev \
|
||||
python3-wheel \
|
||||
python3-venv &&\
|
||||
cd /usr/local/bin &&\
|
||||
ln -s /usr/bin/python3 python &&\
|
||||
ln -s /usr/bin/pip3 pip;
|
||||
|
||||
# Create python3 virtualenv
|
||||
RUN python3 -m venv /opt/venv
|
||||
ENV PATH="/opt/venv/bin:$PATH"
|
||||
|
||||
# Install TensorRT
|
||||
RUN if [ "${CUDA_VERSION:0:2}" = "13" ]; then \
|
||||
wget https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/11.1.0/tars/TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-13.3-Release-external.tar.zst \
|
||||
&& tar -xf TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-13.3-Release-external.tar.zst \
|
||||
&& cp -a TensorRT-11.1.0.106/lib/*.so* /usr/lib/x86_64-linux-gnu/ \
|
||||
&& pip install TensorRT-11.1.0.106/python/tensorrt-11.1.0.106-cp312-none-linux_x86_64.whl ;\
|
||||
elif [ "${CUDA_VERSION:0:2}" = "12" ]; then \
|
||||
wget https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/11.1.0/tars/TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-12.9-Release-external.tar.zst \
|
||||
&& tar -xf TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-12.9-Release-external.tar.zst \
|
||||
&& cp -a TensorRT-11.1.0.106/lib/*.so* /usr/lib/x86_64-linux-gnu/ \
|
||||
&& pip install TensorRT-11.1.0.106/python/tensorrt-11.1.0.106-cp312-none-linux_x86_64.whl ;\
|
||||
else \
|
||||
echo "Invalid CUDA_VERSION"; \
|
||||
exit 1; \
|
||||
fi
|
||||
|
||||
# Install PyPI packages
|
||||
RUN pip3 install --upgrade pip
|
||||
RUN pip3 install setuptools>=41.0.0
|
||||
RUN pip3 install jupyter jupyterlab
|
||||
# Workaround to remove numpy installed with tensorflow
|
||||
RUN pip3 install --upgrade numpy
|
||||
|
||||
# Install Cmake
|
||||
RUN cd /tmp && \
|
||||
wget https://github.com/Kitware/CMake/releases/download/v3.31.11/cmake-3.31.11-Linux-x86_64.sh && \
|
||||
chmod +x cmake-3.31.11-Linux-x86_64.sh && \
|
||||
./cmake-3.31.11-Linux-x86_64.sh --prefix=/usr/local --exclude-subdir --skip-license && \
|
||||
rm ./cmake-3.31.11-Linux-x86_64.sh
|
||||
|
||||
# Install gtest
|
||||
RUN cd /tmp && \
|
||||
git clone https://github.com/google/googletest.git -b v1.14.0 && \
|
||||
cd googletest && \
|
||||
mkdir build && cd build && \
|
||||
cmake .. && \
|
||||
make -j4 && \
|
||||
make install && \
|
||||
rm -rf /tmp/googletest
|
||||
|
||||
# Download NGC client
|
||||
RUN cd /usr/local/bin && wget https://ngc.nvidia.com/downloads/ngccli_cat_linux.zip && unzip ngccli_cat_linux.zip && chmod u+x ngc-cli/ngc && rm ngccli_cat_linux.zip ngc-cli.md5 && echo "no-apikey\nascii\n" | ngc-cli/ngc config set
|
||||
|
||||
# Set environment and working directory
|
||||
ENV TRT_LIBPATH=/usr/lib/x86_64-linux-gnu/
|
||||
ENV TRT_OSSPATH=/workspace/TensorRT
|
||||
ENV PATH="/workspace/TensorRT/build/out:${PATH}:/usr/local/bin/ngc-cli"
|
||||
ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:${TRT_OSSPATH}/build/out:${TRT_LIBPATH}"
|
||||
WORKDIR /workspace
|
||||
|
||||
USER trtuser
|
||||
RUN ["/bin/bash"]
|
||||
@@ -0,0 +1,133 @@
|
||||
#
|
||||
# SPDX-FileCopyrightText: Copyright (c) 1993-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
#
|
||||
|
||||
ARG CUDA_VERSION=13.3.0
|
||||
|
||||
# Multi-arch container support available in non-cudnn containers.
|
||||
FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu26.04
|
||||
LABEL maintainer="NVIDIA CORPORATION"
|
||||
|
||||
ENV TRT_VERSION=11.1.0.106
|
||||
SHELL ["/bin/bash", "-c"]
|
||||
|
||||
# Setup user account and edit default account
|
||||
RUN if id "ubuntu" &>/dev/null; then \
|
||||
usermod -u 1234 ubuntu && \
|
||||
groupmod -g 1234 ubuntu; \
|
||||
fi
|
||||
ARG uid=1000
|
||||
ARG gid=1000
|
||||
RUN groupadd -r -f -g ${gid} trtuser && useradd -o -r -l -u ${uid} -g ${gid} -ms /bin/bash trtuser
|
||||
RUN usermod -aG sudo trtuser
|
||||
RUN echo 'trtuser:nvidia' | chpasswd
|
||||
RUN mkdir -p /workspace && chown trtuser /workspace
|
||||
|
||||
ENV DEBIAN_FRONTEND=noninteractive
|
||||
|
||||
# Install requried libraries
|
||||
RUN apt-get update && apt-get install -y software-properties-common
|
||||
RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||
libcurl4-openssl-dev \
|
||||
wget \
|
||||
git \
|
||||
pkg-config \
|
||||
sudo \
|
||||
ssh \
|
||||
libssl-dev \
|
||||
pbzip2 \
|
||||
pv \
|
||||
bzip2 \
|
||||
unzip \
|
||||
devscripts \
|
||||
lintian \
|
||||
fakeroot \
|
||||
dh-make \
|
||||
build-essential \
|
||||
libnccl2 \
|
||||
libnccl-dev \
|
||||
openmpi-bin \
|
||||
libopenmpi-dev \
|
||||
zstd \
|
||||
ccache
|
||||
|
||||
# Install python3
|
||||
RUN apt-get install -y --no-install-recommends \
|
||||
python3 \
|
||||
python3-pip \
|
||||
python3-dev \
|
||||
python3-wheel \
|
||||
python3-venv &&\
|
||||
cd /usr/local/bin &&\
|
||||
ln -s /usr/bin/python3 python &&\
|
||||
ln -s /usr/bin/pip3 pip;
|
||||
|
||||
# Create python3 virtualenv
|
||||
RUN python3 -m venv /opt/venv
|
||||
ENV PATH="/opt/venv/bin:$PATH"
|
||||
|
||||
# Install TensorRT
|
||||
RUN if [ "${CUDA_VERSION:0:2}" = "13" ]; then \
|
||||
wget https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/11.1.0/tars/TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-13.3-Release-external.tar.zst \
|
||||
&& tar -xf TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-13.3-Release-external.tar.zst \
|
||||
&& cp -a TensorRT-11.1.0.106/lib/*.so* /usr/lib/x86_64-linux-gnu/ \
|
||||
&& pip install TensorRT-11.1.0.106/python/tensorrt-11.1.0.106-cp314-none-linux_x86_64.whl ;\
|
||||
elif [ "${CUDA_VERSION:0:2}" = "12" ]; then \
|
||||
wget https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/11.1.0/tars/TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-12.9-Release-external.tar.zst \
|
||||
&& tar -xf TensorRT-Enterprise-11.1.0.106-Linux-x86_64-cuda-12.9-Release-external.tar.zst \
|
||||
&& cp -a TensorRT-11.1.0.106/lib/*.so* /usr/lib/x86_64-linux-gnu/ \
|
||||
&& pip install TensorRT-11.1.0.106/python/tensorrt-11.1.0.106-cp314-none-linux_x86_64.whl ;\
|
||||
else \
|
||||
echo "Invalid CUDA_VERSION"; \
|
||||
exit 1; \
|
||||
fi
|
||||
|
||||
# Install PyPI packages
|
||||
RUN pip3 install --upgrade pip
|
||||
RUN pip3 install setuptools>=41.0.0
|
||||
RUN pip3 install jupyter jupyterlab
|
||||
# Workaround to remove numpy installed with tensorflow
|
||||
RUN pip3 install --upgrade numpy
|
||||
|
||||
# Install Cmake
|
||||
RUN cd /tmp && \
|
||||
wget https://github.com/Kitware/CMake/releases/download/v3.31.11/cmake-3.31.11-Linux-x86_64.sh && \
|
||||
chmod +x cmake-3.31.11-Linux-x86_64.sh && \
|
||||
./cmake-3.31.11-Linux-x86_64.sh --prefix=/usr/local --exclude-subdir --skip-license && \
|
||||
rm ./cmake-3.31.11-Linux-x86_64.sh
|
||||
|
||||
# Install gtest
|
||||
RUN cd /tmp && \
|
||||
git clone https://github.com/google/googletest.git -b v1.14.0 && \
|
||||
cd googletest && \
|
||||
mkdir build && cd build && \
|
||||
cmake .. && \
|
||||
make -j4 && \
|
||||
make install && \
|
||||
rm -rf /tmp/googletest
|
||||
|
||||
# Download NGC client
|
||||
RUN cd /usr/local/bin && wget https://ngc.nvidia.com/downloads/ngccli_cat_linux.zip && unzip ngccli_cat_linux.zip && chmod u+x ngc-cli/ngc && rm ngccli_cat_linux.zip ngc-cli.md5 && echo "no-apikey\nascii\n" | ngc-cli/ngc config set
|
||||
|
||||
# Set environment and working directory
|
||||
ENV TRT_LIBPATH=/usr/lib/x86_64-linux-gnu/
|
||||
ENV TRT_OSSPATH=/workspace/TensorRT
|
||||
ENV PATH="/workspace/TensorRT/build/out:${PATH}:/usr/local/bin/ngc-cli"
|
||||
ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:${TRT_OSSPATH}/build/out:${TRT_LIBPATH}"
|
||||
WORKDIR /workspace
|
||||
|
||||
USER trtuser
|
||||
RUN ["/bin/bash"]
|
||||
@@ -0,0 +1,115 @@
|
||||
#
|
||||
# SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
#
|
||||
|
||||
ARG CUDA_VERSION=13.3.0
|
||||
ARG OS_VERSION=24.04
|
||||
|
||||
FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${OS_VERSION}
|
||||
LABEL maintainer="NVIDIA CORPORATION"
|
||||
|
||||
ENV TRT_VERSION 11.1.0.106
|
||||
ENV DEBIAN_FRONTEND=noninteractive
|
||||
|
||||
# Setup user account and edit default account
|
||||
RUN if id "ubuntu" &>/dev/null; then \
|
||||
usermod -u 1234 ubuntu && \
|
||||
groupmod -g 1234 ubuntu; \
|
||||
fi
|
||||
ARG uid=1000
|
||||
ARG gid=1000
|
||||
RUN groupadd -r -f -g ${gid} trtuser && useradd -o -r -l -u ${uid} -g ${gid} -ms /bin/bash trtuser
|
||||
RUN usermod -aG sudo trtuser
|
||||
RUN echo 'trtuser:nvidia' | chpasswd
|
||||
RUN mkdir -p /workspace && chown trtuser /workspace
|
||||
|
||||
# Install requried libraries + aarch64 toolchains
|
||||
RUN apt-get update && apt-get install -y software-properties-common
|
||||
RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||
libcurl4-openssl-dev \
|
||||
wget \
|
||||
git \
|
||||
pkg-config \
|
||||
sudo \
|
||||
ssh \
|
||||
pbzip2 \
|
||||
pv \
|
||||
bzip2 \
|
||||
unzip \
|
||||
build-essential \
|
||||
g++-aarch64-linux-gnu \
|
||||
libnccl2 \
|
||||
libnccl-dev \
|
||||
openmpi-bin \
|
||||
libopenmpi-dev \
|
||||
zstd \
|
||||
ccache
|
||||
|
||||
# Install python3
|
||||
RUN apt-get install -y --no-install-recommends \
|
||||
python3 \
|
||||
python3-pip \
|
||||
python3-dev \
|
||||
python3-wheel \
|
||||
python3-venv &&\
|
||||
cd /usr/local/bin &&\
|
||||
ln -s /usr/bin/python3 python &&\
|
||||
ln -s /usr/bin/pip3 pip;
|
||||
|
||||
# Create python3 virtualenv
|
||||
RUN python3 -m venv /opt/venv
|
||||
ENV PATH="/opt/venv/bin:$PATH"
|
||||
|
||||
# Install Cmake
|
||||
RUN cd /tmp && \
|
||||
wget https://github.com/Kitware/CMake/releases/download/v3.31.11/cmake-3.31.11-Linux-x86_64.sh && \
|
||||
chmod +x cmake-3.31.11-Linux-x86_64.sh && \
|
||||
./cmake-3.31.11-Linux-x86_64.sh --prefix=/usr/local --exclude-subdir --skip-license && \
|
||||
rm ./cmake-3.31.11-Linux-x86_64.sh
|
||||
|
||||
# Install gtest
|
||||
RUN cd /tmp && \
|
||||
git clone https://github.com/google/googletest.git -b v1.14.0 && \
|
||||
cd googletest && \
|
||||
mkdir build && cd build && \
|
||||
cmake .. && \
|
||||
make -j4 && \
|
||||
make install && \
|
||||
rm -rf /tmp/googletest
|
||||
|
||||
# Install CUDA cross compile toolchain
|
||||
RUN wget https://developer.download.nvidia.com/compute/cuda/13.3.0/local_installers/cuda-repo-cross-sbsa-ubuntu2404-13-3-local_13.3.0-1_all.deb && \
|
||||
dpkg -i cuda-repo-cross-sbsa-ubuntu2404-13-3-local_13.3.0-1_all.deb && \
|
||||
cp /var/cuda-repo-cross-sbsa-ubuntu2404-13-3-local/cuda-*-keyring.gpg /usr/share/keyrings/ && \
|
||||
apt-get update && \
|
||||
apt-get -y install cuda-cross-sbsa-13-3
|
||||
|
||||
# Unpack libnvinfer.
|
||||
|
||||
RUN wget https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/11.1.0/tars/TensorRT-Enterprise-11.1.0.106-Linux-aarch64-cuda-13.3-Release-external.tar.zst && \
|
||||
tar -xf TensorRT-Enterprise-11.1.0.106-Linux-aarch64-cuda-13.3-Release-external.tar.zst && \
|
||||
cp -a TensorRT-11.1.0.106/lib/*.so* /usr/lib/aarch64-linux-gnu
|
||||
|
||||
# Link required library
|
||||
RUN cd /usr/aarch64-linux-gnu/lib && ln -sf librt.so.1 librt.so
|
||||
|
||||
# Set environment and working directory
|
||||
ENV TRT_LIBPATH /usr/lib/aarch64-linux-gnu
|
||||
ENV TRT_OSSPATH /workspace/TensorRT
|
||||
WORKDIR /workspace
|
||||
|
||||
USER trtuser
|
||||
RUN ["/bin/bash"]
|
||||
Reference in New Issue
Block a user