NVIDIA
TensorRT LLM Develop
Container
NVIDIA
TensorRT LLM Develop

TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs.

  • LayerLabelCreated
    sha256:f0e555f1232128facfa8dbd6a43b750f98265b625ff0bbae753aaeda36b880fdRUN
    GITHUB_MIRROR=https://urm.nvidia.com/artifactory/github-go-remote PYTHON_VERSION=3.12.3 TRT_VER= CUDA_VER= CUDNN_VER= NCCL_VER= CUBLAS_VER= TORCH_INSTALL_TYPE=skip /bin/bash -c pip3 install --upgrade --no-cache-dir "h11>=0.16" "tornado>=6.5.0" "setuptools>=78.1.1,<80"
    06/09/2025 4:48 PM UTC
    sha256:ec2bbc6525f02990c65498cc61db34a35bf127ae4d088be5cd2e97618ce741c2RUN
    GITHUB_MIRROR=https://urm.nvidia.com/artifactory/github-go-remote PYTHON_VERSION=3.12.3 TRT_VER= CUDA_VER= CUDNN_VER= NCCL_VER= CUBLAS_VER= TORCH_INSTALL_TYPE=skip /bin/bash -c pip3 install opencv-python-headless --force-reinstall --no-deps --no-cache-dir
    06/09/2025 4:48 PM UTC
    sha256:5291396a6cff7b2fbf341fcbe69a91798227ba1322ea14b63a02429484ba56f0RUN
    GITHUB_MIRROR=https://urm.nvidia.com/artifactory/github-go-remote PYTHON_VERSION=3.12.3 TRT_VER= CUDA_VER= CUDNN_VER= NCCL_VER= CUBLAS_VER= TORCH_INSTALL_TYPE=skip /bin/bash -c pip3 uninstall -y opencv &&
      rm -rf /usr/local/lib/python3*/dist-packages/cv2/
    06/09/2025 4:48 PM UTC
    sha256:8cae29f2be2adaefaa6b4a9d2d73ede86845fefdb2e8ab111498d6c88b81d59cRUN
    GITHUB_MIRROR=https://urm.nvidia.com/artifactory/github-go-remote PYTHON_VERSION=3.12.3 TRT_VER= CUDA_VER= CUDNN_VER= NCCL_VER= CUBLAS_VER= TORCH_INSTALL_TYPE=skip /bin/bash -c bash ./install_pytorch.sh $TORCH_INSTALL_TYPE &&
      rm install_pytorch.sh
    06/09/2025 4:48 PM UTC
    sha256:c454945ce74afc282e2ce7ca5f01a6fb437756708d6c69d6349215c9a1847688COPY
    docker/common/install_pytorch.sh install_pytorch.sh
    06/09/2025 4:48 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    TORCH_INSTALL_TYPE=skip
    06/09/2025 4:48 PM UTC
    sha256:2b6d32217b2f69c6726ffa00112191dc3b4fee710df1809b97a6a3f3efd19e41RUN
    GITHUB_MIRROR=https://urm.nvidia.com/artifactory/github-go-remote PYTHON_VERSION=3.12.3 TRT_VER= CUDA_VER= CUDNN_VER= NCCL_VER= CUBLAS_VER= /bin/bash -c bash ./install_mpi4py.sh &&
      rm install_mpi4py.sh
    06/09/2025 4:48 PM UTC
    sha256:abab1505e3f000f7ba8b4e0c7fc459b2aa965c015aa04a46c3bd3a974f41b7b2COPY
    docker/common/install_mpi4py.sh install_mpi4py.sh
    06/09/2025 4:48 PM UTC
    sha256:fa348759acec66ee24471dd90d27a9e973bc83b845d9e2b43d6814d53b25d856RUN
    GITHUB_MIRROR=https://urm.nvidia.com/artifactory/github-go-remote PYTHON_VERSION=3.12.3 TRT_VER= CUDA_VER= CUDNN_VER= NCCL_VER= CUBLAS_VER= /bin/bash -c bash ./install_polygraphy.sh &&
      rm install_polygraphy.sh
    06/09/2025 4:48 PM UTC
    sha256:641bb655fa1495de44e4ee26384ee47616813950bc801151cdfb51dffa3712ddCOPY
    docker/common/install_polygraphy.sh install_polygraphy.sh
    06/09/2025 4:46 PM UTC
    ...