New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

No module named 'cuda' #130

Open

AdamSEY opened this issue Jun 17, 2024 · 1 comment

AdamSEY commented Jun 17, 2024 •

edited

Loading

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'

Owner

Linaom1214 commented Jun 17, 2024

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment