search

triple-Mu / YOLOv8-TensorRT

YOLOv8 using TensorRT accelerate !
MIT License
1.23k stars 210 forks source link

No module named 'onnx' #191

Closed ugkjkj closed 1 month ago

ugkjkj commented 6 months ago

C:\Users\15258>python3 "C:\Users\15258\Downloads\YOLOv8-TensorRT-main\YOLOv8-TensorRT-main\infer-det.py" --engine "C:\Users\15258\Desktop\model++\runs\detect\train26\weights\best.engine" --imgs "C:\Users\15258\Desktop\model++\000002 copy.jpg" --out-dir outputs --device cuda:0 Traceback (most recent call last): File "C:\Users\15258\Downloads\YOLOv8-TensorRT-main\YOLOv8-TensorRT-main\infer-det.py", line 1, in from models import TRTModule # isort:skip File "C:\Users\15258\Downloads\YOLOv8-TensorRT-main\YOLOv8-TensorRT-main\models__init__.py", line 1, in from .engine import EngineBuilder, TRTModule, TRTProfilerV0, TRTProfilerV1 # isort:skip # noqa: E501 File "C:\Users\15258\Downloads\YOLOv8-TensorRT-main\YOLOv8-TensorRT-main\models\engine.py", line 7, in import onnx ModuleNotFoundError: No module named 'onnx'

triple-Mu commented 6 months ago

C:\Users\15258>python3 "C:\Users\15258\Downloads\YOLOv8-TensorRT-main\YOLOv8-TensorRT-main\infer-det.py" --engine "C:\Users\15258\Desktop\model++\runs\detect\train26\weights\best.engine" --imgs "C:\Users\15258\Desktop\model++\000002 copy.jpg" --out-dir outputs --device cuda:0 Traceback (most recent call last): File "C:\Users\15258\Downloads\YOLOv8-TensorRT-main\YOLOv8-TensorRT-main\infer-det.py", line 1, in from models import TRTModule # isort:skip File "C:\Users\15258\Downloads\YOLOv8-TensorRT-main\YOLOv8-TensorRT-main\modelsinit.py", line 1, in from .engine import EngineBuilder, TRTModule, TRTProfilerV0, TRTProfilerV1 # isort:skip # noqa: E501 File "C:\Users\15258\Downloads\YOLOv8-TensorRT-main\YOLOv8-TensorRT-main\models\engine.py", line 7, in import onnx ModuleNotFoundError: No module named 'onnx'

pip install onnx

ugkjkj commented 6 months ago

(model) C:\Users\15258>pip install onnx Requirement already satisfied: onnx in d:\anaconda\envs\model\lib\site-packages (1.15.0) Requirement already satisfied: numpy in d:\anaconda\envs\model\lib\site-packages (from onnx) (1.26.2) Requirement already satisfied: protobuf>=3.20.2 in d:\anaconda\envs\model\lib\site-packages (from onnx) (4.25.1)

triple-Mu commented 6 months ago

python -m pip install onnx

ugkjkj commented 6 months ago

(model) C:\Users\15258>python -m pip install onnx Requirement already satisfied: onnx in d:\anaconda\envs\model\lib\site-packages (1.15.0) Requirement already satisfied: numpy in d:\anaconda\envs\model\lib\site-packages (from onnx) (1.26.2) Requirement already satisfied: protobuf>=3.20.2 in d:\anaconda\envs\model\lib\site-packages (from onnx) (4.25.1)

ugkjkj commented 6 months ago

Traceback (most recent call last): File "C:\Users\15258\Downloads\YOLOv8-TensorRT-main\YOLOv8-TensorRT-main\infer-det.py", line 1, in from models import TRTModule # isort:skip File "C:\Users\15258\Downloads\YOLOv8-TensorRT-main\YOLOv8-TensorRT-main\models__init__.py", line 1, in from .engine import EngineBuilder, TRTModule, TRTProfilerV0, TRTProfilerV1 # isort:skip # noqa: E501 File "C:\Users\15258\Downloads\YOLOv8-TensorRT-main\YOLOv8-TensorRT-main\models\engine.py", line 7, in import onnx ModuleNotFoundError: No module named 'onnx'

triple-Mu commented 6 months ago

how about uninstall onnx and install then.