Error: 'from cuda import cudart'

platform:NVIDIA TX2 NX
Environment:jetpack4.6.3 python3.6 cuda10.2 pycuda2020.1
error:“no module named cuda”

you need to install this.

cuda 10.2 has in this venv(look at the second photo)

It’s not the same thing. Installing CUDA so that you can run nvcc -V is not the same thing as installing the CUDA python bindings I indicated.

cuda-python requires at least py3.7,mabey i can try pycuda. Do you have a case for using pycuda to trt inference.