Sign inSign up

openeuler/tensorrt-llm

Sponsored OSS

By openeuler

Updated 2 months ago

Image
0

6.4K

openeuler/tensorrt-llm repository overview

Quick reference

TensorRT-LLM | openEuler

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. TensorRT LLM also contains components to create Python and C++ runtimes that orchestrate the inference execution in a performant way.

The tag of each TensorRT-LLM docker image is consist of the version of TensorRT-LLM and the version of basic image. The details are as follows

TagsCurrentlyArchitectures
1.2.1-oe2403sp4tensorrt-llm 1.2.1 on openEuler 24.03-lts-sp4amd64, arm64
1.2.1-oe2403sp3tensorrt-llm 1.2.1 on openEuler 24.03-lts-sp3amd64, arm64

Usage

This image provides the TensorRT LLM inference environment with CUDA 13.1 and PyTorch 2.9.To run the container with GPU support:

docker run -it --gpus all --name trtllm-dev openeuler/tensorrt-llm:{Tag}

Once inside the container, verify the installation:

python3 -c "from tensorrt_llm import LLM; print('TensorRT LLM ready')"

Question and answering

If you have any questions or want to use some special features, please submit an issue or a pull request on openeuler-docker-images.

Tag summary

Content type

Image

Digest

sha256:7f2b8fbf2

Size

9.1 GB

Last updated

2 months ago

docker pull openeuler/tensorrt-llm

This week's pulls

Pulls:

11

Last week