安装环境:
JetPack 4.6.1-4.6.4
配置cuda 环境
$ sudo gedit ~/.bashrc
export PATH=/usr/local/cuda-10.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.2
$ source ~/.bashrc
$ nvcc -V 检测一下是否配置成功
下载pycuda源码
下载完之后解压,进入解压出来的文件
tar zxvf pycuda-2022.1.tar.gz
cd pycuda-2022.1/
python3 configure.py --cuda-root=/usr/local/cuda-10.2
sudo python3 setup.py install
参考:
https://blog.csdn.net/weixin_44501699/article/details/106470671