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:b917503a705d83777313948c6e016997afb2f0cfd9f0c9b34f2bfeacc9fe304dRUN
    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" "protobuf>=4.25.8" "jupyter-core>=5.8.1"
    07/19/2025 5:46 PM UTC
    sha256:9ce333ada0beb679994eadb874ee1d559fc980a49b53940ed1d478f8e69b59a0RUN
    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
    07/19/2025 5:46 PM UTC
    sha256:2b999f52022bd3b877c29dabce463798fc671b6a3e6d08f93d9222524a348b92RUN
    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/
    07/19/2025 5:44 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    PYTORCH_CUDA_ALLOC_CONF=garbage_collection_threshold:0.99999
    07/19/2025 5:44 PM UTC
    sha256:643d444886c79740723f44ebabd13b67b02c5219ae482a6c94fd9db9cde080d9RUN
    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
    07/19/2025 5:44 PM UTC
    sha256:7a40b3d76756131bdfa3e1cad5797b10d0158a1e381d6a61d5fa8455e33f86c4COPY
    docker/common/install_pytorch.sh install_pytorch.sh
    07/19/2025 5:44 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    TORCH_INSTALL_TYPE=skip
    07/19/2025 5:44 PM UTC
    sha256:bbacf9f93729038d18409f8e8c62f1c2ec74644fcbb9a19313a54171d80d34a6RUN
    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 GITHUB_MIRROR=$GITHUB_MIRROR bash ./install_mpi4py.sh &&
      rm install_mpi4py.sh
    07/19/2025 5:44 PM UTC
    sha256:696d8218b758ebd9633059b5243cb95f1f4713d4c44c0e6be57923e2e2492948COPY
    docker/common/install_mpi4py.sh install_mpi4py.sh
    07/19/2025 5:42 PM UTC
    sha256:94660bbd6bf44645c46765f1db13fc67e743ba72743502ed1a64ea687460c8faRUN
    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
    07/19/2025 5:42 PM UTC
    ...