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:94c37ce56b02636722df9eea47d1392606a214e22869effc72abbf18d784e442RUN
    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/21/2025 8:35 AM UTC
    sha256:a70043e63486ae6c2093fce6a127d7548bd2f8e782092a712c575d470f2e9aaaRUN
    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_deep_ep.sh &&
      rm install_deep_ep.sh
    06/21/2025 8:35 AM UTC
    sha256:8d01ac221736018683cac5ff0b67d09e4b430671644ee18c213ebe03a0cdd417COPY
    docker/common/install_deep_ep.sh install_deep_ep.sh
    06/21/2025 8:31 AM UTC
    sha256:a1ae05027d3e911a219679adf7944f4562d7bd8569593ac28bf954accda95c06RUN
    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/21/2025 8:31 AM UTC
    sha256:8f4e35a0d7c9eab8affc3c1285b689642a5c59acfb468fc4454f603e2ae2eed3RUN
    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/21/2025 8:29 AM UTC
    sha256:aba2bf96aa6cced1a20dd53dc713f5c0f2dffe8731e6b82a97823e0ce0bccfa6RUN
    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/21/2025 8:29 AM UTC
    sha256:5223862e28b99cdb4a6db6ed5da3da08e3942668e5ea4fc76f2554ebd6dcf715COPY
    docker/common/install_pytorch.sh install_pytorch.sh
    06/21/2025 8:29 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    TORCH_INSTALL_TYPE=skip
    06/21/2025 8:29 AM UTC
    sha256:4a39509075cc84f0763369d6b47cdb0cad05e6ded357e9cc37f5323a62884f5cRUN
    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/21/2025 8:29 AM UTC
    sha256:e0b1a6383afbbce4fe8bed5ab1443f75ff7cecf23e3356463e3c2021dfc99c8dCOPY
    docker/common/install_mpi4py.sh install_mpi4py.sh
    06/21/2025 8:26 AM UTC
    ...