search

TrojanXu / yolov5-tensorrt

A tensorrt implementation of yolov5: https://github.com/ultralytics/yolov5
Apache License 2.0
191 stars 46 forks source link

tensorrt installation #20

Closed 123456789mojtaba closed 3 years ago

123456789mojtaba commented 3 years ago

how can I install tensorrt in pytorch in anaconda?

TrojanXu commented 3 years ago

tensorrt python binding is not part of PyTorch. If you have the tensorrt package, install from the python wheel from path_to_tensorrt_package/ptyhon/pyverxxx.whl. Otherwise, you can easily reproduce results with docker image pytorch:20.01-py3 from ngc.nvidia.com

123456789mojtaba commented 3 years ago

can we convert yolov5 to tensorrt? I read it has upsampling layer and leaky relu that have problem with tensorrt?what should we do? plz help

On Wed, Jul 8, 2020 at 5:20 AM autoregister notifications@github.com wrote:

Closed #20 https://github.com/TrojanXu/yolov5-tensorrt/issues/20.

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub https://github.com/TrojanXu/yolov5-tensorrt/issues/20#event-3522878368, or unsubscribe https://github.com/notifications/unsubscribe-auth/AM3F7254X7SKV3RV2LP7K2LR2O7FNANCNFSM4OTNPUAQ .