Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/kernferm/pytorch-to-tensorrt-model-converter
This project provides a comprehensive Python script to convert a PyTorch model to an ONNX model and then to a TensorRT engine for NVIDIA GPUs, followed by performing inference using the TensorRT engine. This script is designed to handle the entire conversion process seamlessly.
https://github.com/kernferm/pytorch-to-tensorrt-model-converter
convert gpu numpy nvidia nvidia-gpu onnx pytorch tensorrt tensorrt-conversion yolo yolov5 yolov8
Last synced: about 1 month ago
JSON representation
This project provides a comprehensive Python script to convert a PyTorch model to an ONNX model and then to a TensorRT engine for NVIDIA GPUs, followed by performing inference using the TensorRT engine. This script is designed to handle the entire conversion process seamlessly.
- Host: GitHub
- URL: https://github.com/kernferm/pytorch-to-tensorrt-model-converter
- Owner: KernFerm
- License: apache-2.0
- Created: 2024-07-16T18:13:49.000Z (6 months ago)
- Default Branch: main
- Last Pushed: 2024-09-10T08:28:02.000Z (5 months ago)
- Last Synced: 2024-09-10T10:45:22.872Z (5 months ago)
- Topics: convert, gpu, numpy, nvidia, nvidia-gpu, onnx, pytorch, tensorrt, tensorrt-conversion, yolo, yolov5, yolov8
- Language: Python
- Homepage:
- Size: 112 KB
- Stars: 1
- Watchers: 2
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.MD
- License: LICENSE
- Code of conduct: CODE_OF_CONDUCT.md