Skip to content

Add TensorRT dependencies#8553

Merged
glenn-jocher merged 1 commit intoultralytics:masterfrom
Zengyf-CVer:patch-10
Jul 12, 2022
Merged

Add TensorRT dependencies#8553
glenn-jocher merged 1 commit intoultralytics:masterfrom
Zengyf-CVer:patch-10

Conversation

@Zengyf-CVer
Copy link
Copy Markdown
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
Copy Markdown
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.

2 participants