Pycuda error on RTX 3060

Accelerated Computing CUDA CUDA Setup and Installation
, , , ,
1

Hi everyone,
I am getting error when I running yolo.trt file how can I fix that.

Error:

Traceback (most recent call last):
  File "yolo_trt.py", line 8, in <module>
    from utils.yolo_with_plugins import TrtYOLO
  File "/home/oret/tensorrt_demos/utils/yolo_with_plugins.py", line 14, in <module>
    import pycuda.driver as cuda
ModuleNotFoundError: No module named 'pycuda'
2

install pycuda

It’s not a nvidia-maintained product, but you can find instructions with some searches.