TensorRT is a high-performance deep learning inference optimizer and runtime library developed by NVIDIA. It is designed to optimize and accelerate inference for deep learning models deployed on NVIDIA GPUs.
TensorRT works by optimizing the computation graph of a deep learning model for efficient execution on the GPU. It achieves this by fusing multiple layers into a single operation, pruning unused operations, and selecting the most efficient algorithms for each operation.
TensorRT supports various deep learning frameworks, including TensorFlow, PyTorch, and ONNX, and it can be integrated into a wide range of applications, including autonomous vehicles, robotics, natural language processing, and computer vision.
By using TensorRT, developers can achieve significant speedups in inference time compared to running the same models on a CPU. This makes it an essential tool for real-time and low-latency applications that require high-performance inference.
$ git clone
Content type
Image
Digest
sha256:7c1c4ab21…
Size
10.3 GB
Last updated
over 3 years ago
docker pull simrankang/tensor-rt