No module named 'cuda'

[AdamSEY]

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]

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

[NadimDib]

@AdamSEY if pip install cuda-python does not work for you it's probably because you are using a jetson device like the AGX Xavier (like me) or so. pip3 will not be able to find a compatible cuda-python version so you should downgrade to a slightly older one. I used pip3 install cuda-python==11.5.0 and it worked. It's going to depend on your Jetpack version (I have 5.1.5). If that doesn't work try to change the versions and iterate over them. It's a shame really :(