search

TrojanXu / yolov5-tensorrt

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

No module named 'tensorrt' #29

Closed WestbrookZero closed 4 years ago

WestbrookZero commented 4 years ago

My environment is Win10，already install uff-0.6.5-py2.py3-none-any.whl and graphsurgeon-0.4.1-py2.py3-none-any.whl,but still have this error: Traceback (most recent call last): File "D:/Projects/DeepLearning/yolov5-tensorrt/main.py", line 8, in import tensorrt as trt ModuleNotFoundError: No module named 'tensorrt'

TrojanXu commented 4 years ago

You need to install tensorrt python wheel contained in downloaded tensorrt tar package.

WestbrookZero commented 4 years ago

there is no tensorrt whl in win10 package

TrojanXu commented 4 years ago

On windows, you might need WSL but with a lot of performance drop. Or just find a linux machine...