JetPack4.4版本
使用之前配置cuda的环境
之后下载​​pycuda​​源码
下载完之后解压
进入解压出来的文件
tar zxvf pycuda-2019.1.2.tar.gz    
cd pycuda-2019.1.2/
python3 configure.py --cuda-root=/usr/local/cuda-10.2
sudo python3 setup.py install

​​原文链接​​