Installing on Linux

1. Install the NVIDIA Container Toolkit.

2. Install TensorRT-LLM.

   # Obtain and start the basic docker image environment (optional).
   docker run --rm --runtime=nvidia --gpus all --entrypoint /bin/bash -it nvidia/cuda:12.1.0-devel-ubuntu22.04
   
   # Install dependencies, TensorRT-LLM requires Python 3.10
   apt-get update && apt-get -y install python3.10 python3-pip openmpi-bin libopenmpi-dev git
   
   # Install the latest preview version (corresponding to the main branch) of TensorRT-LLM.
   # If you want to install the stable version (corresponding to the release branch), please
   # remove the `--pre` option.
   pip3 install tensorrt_llm -U --pre --extra-index-url https://pypi.nvidia.com
   
   # Check installation
   python3 -c "import tensorrt_llm"
   

3. Install the requirements inside the Docker container.

   git clone https://github.com/NVIDIA/TensorRT-LLM.git
   cd TensorRT-LLM
   pip install -r examples/bloom/requirements.txt
   git lfs install
   

Beyond the local execution, you can also use the NVIDIA Triton Inference Server to create a production-ready deployment of your LLM as described in this Optimizing Inference on Large Language Models with NVIDIA TensorRT-LLM blog.