search

Linaom1214 / TensorRT-For-YOLO-Series

tensorrt for yolo series (YOLOv10,YOLOv9,YOLOv8,YOLOv7,YOLOv6,YOLOX,YOLOv5), nms plugin support
877 stars 151 forks source link

No module named 'cuda' #130

Open AdamSEY opened 2 months ago

AdamSEY commented 2 months ago

pip install cuda-python Requirement already satisfied: cuda-python

python export.py -o yolov10n.onnx -e yolov10.trt --end2end --v10 -p fp32

File "TensorRT-For-YOLO-Series-cuda-python\export.py", line 8, in from cuda import cudart ModuleNotFoundError: No module named 'cuda'

Linaom1214 commented 2 months ago

pip install cuda-python Requirement already satisfied: cuda-python

python export.py -o yolov10n.onnx -e yolov10.trt --end2end --v10 -p fp32

File "TensorRT-For-YOLO-Series-cuda-python\export.py", line 8, in from cuda import cudart ModuleNotFoundError: No module named 'cuda'

install cuda-python and tensorrt

pip install tensorrt pip install cuda-python