TensorRT
mmrazor
TensorRT | mmrazor | |
---|---|---|
5 | 4 | |
2,340 | 1,367 | |
1.8% | 1.4% | |
9.5 | 2.8 | |
7 days ago | 21 days ago | |
Python | Python | |
BSD 3-clause "New" or "Revised" License | Apache License 2.0 |
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.
TensorRT
- Learn TensorRT optimization
- I made TensorRT example. I hope this will help beginners. And I also have a question about TensorRT best practice.
- [P] [D] I made TensorRT example. I hope this will help beginners. And I also have a question about TensorRT best practice.
-
[P] 4.5 times faster Hugging Face transformer inference by modifying some Python AST
Have you tried the new Torch-TensorRT compiler from NVIDIA?
-
PyTorch 1.10
You can quantize your model to FP16 or Int8 using PTQ as well and it should give you an additional speed up inference wise.
Here is a tutorial[2] to leverage TRTorch.
[1] https://github.com/NVIDIA/TRTorch/tree/master/core
mmrazor
-
MMDeploy: Deploy All the Algorithms of OpenMMLab
MMRazor: OpenMMLab model compression toolbox and benchmark.
- Still worrying about model compression? MMRazor may work for you.
- Still worrying about model compression? MMRazor is all you need
-
[P] 4.5 times faster Hugging Face transformer inference by modifying some Python AST
https://github.com/open-mmlab/mmrazor ,it may work for you~
What are some alternatives?
torch2trt - An easy to use PyTorch to TensorRT converter
transformer-deploy - Efficient, scalable and enterprise-grade CPU/GPU inference server for 🤗 Hugging Face transformer models 🚀
onnxruntime - ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator
neural-compressor - SOTA low-bit LLM quantization (INT8/FP8/INT4/FP4/NF4) & sparsity; leading model compression techniques on TensorFlow, PyTorch, and ONNX Runtime
cutlass - CUDA Templates for Linear Algebra Subroutines
mmaction2 - OpenMMLab's Next Generation Video Understanding Toolbox and Benchmark
onnx-simplifier - Simplify your onnx model
Pointnet_Pointnet2_pytorch - PointNet and PointNet++ implemented by pytorch (pure python) and on ModelNet, ShapeNet and S3DIS.
TensorRT - NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.
PaddleViT - :robot: PaddleViT: State-of-the-art Visual Transformer and MLP Models for PaddlePaddle 2.0+
ttach - Image Test Time Augmentation with PyTorch!