ModuleNotFoundError: No module named 'pycuda._driver

[father]

Traceback (most recent call last):
File “demo.py”, line 2, in
import pycuda.autoinit
File “/home/ljx/pycuda2/pycuda-2021.1/pycuda/autoinit.py”, line 1, in
import pycuda.driver as cuda
File “/home/ljx/pycuda2/pycuda-2021.1/pycuda/driver.py”, line 65, in
from pycuda._driver import * # noqa
ModuleNotFoundError: No module named ‘pycuda._driver’

[father]

jetpack4.6.1

[AastaLLL]

Hi,

Have you installed the pycuda library first?
You can do this with the following command:

$ export PATH=/usr/local/cuda-10.2/bin:$PATH
$ export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH
$ pip3 install pycuda --user

Thanks.

[father]

OK, I think I’ve solved it. He can’t compile under the path of source code installation. If he writes the program under another path again, he won’t report an error

[AastaLLL]

Good to know this.
Thanks for the feedback.