ModuleNotFoundError: No module named 'cuda'

[sssssshf]

Description

Environment

TensorRT Version:

NVIDIA GPU:

NVIDIA Driver Version:

CUDA Version:

CUDNN Version:

Operating System:

Python Version (if applicable):

Tensorflow Version (if applicable):

PyTorch Version (if applicable):

Baremetal or Container (if so, version):

Relevant Files

Model link:

Steps To Reproduce

Commands or scripts:

Have you tried the latest release?:

Can this model run on other frameworks? For example run ONNX model with ONNXRuntime (polygraphy run <model.onnx> --onnxrt):

[sssssshf]

ModuleNotFoundError: No module named 'cuda'

when i run
TensorRT-release-10.0/samples/python/efficientnet/infer.py
how to install that libs ?

[lix19937]

Maybe your cuda not install right.
Use follow cmd

python -c "from cuda import cuda;print(cuda.CUresult.CUDA_SUCCESS)"
python -c "from cuda import cudart;print(cudart.cudaMalloc(10))"   

[dilwolf]

I faced the same error when I was converting my custom YOLO model to TensorRT engine. I just run below command, and it solved this error. I hope this will help:

pip3 install cuda-python