Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add TensorRT dependencies #8553

Merged
merged 1 commit into from Jul 12, 2022
Merged

Conversation

Zengyf-CVer
Copy link
Contributor

@Zengyf-CVer Zengyf-CVer commented Jul 12, 2022

When executing the command python export.py --weights yolov5s.pt --include engine --device 0, you need to install nvidia-pyindex and nvidia-tensorrt to export yolov5s.engine .

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhanced support for NVIDIA TensorRT in YOLOv5 by updating dependencies.

πŸ“Š Key Changes

  • Added nvidia-pyindex as a new requirement.
  • Added nvidia-tensorrt as a new requirement.

🎯 Purpose & Impact

  • Purpose: These changes aim to facilitate the export of YOLOv5 models to TensorRT format, which is designed for efficient inference on NVIDIA GPUs.
  • Impact: Users looking to deploy YOLOv5 on NVIDIA hardware can now do so more seamlessly, potentially improving the performance and speed of their models in production. πŸš€

@glenn-jocher glenn-jocher merged commit fbd3020 into ultralytics:master Jul 12, 2022
@glenn-jocher
Copy link
Member

@Zengyf-CVer PR is merged. Thank you for your contributions to YOLOv5 πŸš€ and Vision AI ⭐

@glenn-jocher glenn-jocher self-assigned this Jul 12, 2022
ctjanuhowski pushed a commit to ctjanuhowski/yolov5 that referenced this pull request Sep 8, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants