jetson 安装 pycuda

利用源码编译方式：

pycuda 官方下载地址https://pypi.org/project/pycuda/#history

1. 下载相应压缩包，我选择的最新版本，解压：

sudo tar xvf  pycuda-2023.1.tar.gz

进入解压后的文件夹中：

cd pycuda-2023.1

2. 看一下cuda 是否运行正常，利用下面命令：

nvcc --version

查看cuda 版本，若出现版本和位置，则继续；如果报错就需要配置一下cuda 的环境变量:

进行cuda的环境配置：

sudo gedit ~/.bashrc

打开文件后，在最后添加如下信息：（根据自己cuda版本进行编辑）

export PATH="/usr/local/cuda-11.4/bin:$PATH"

export LD_LIBRARY_PATH="/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH"

关闭文件，使其生效：

source ~/.bashrc

最后再输入nvcc --version即可查看nvcc版本，成功就可以进行第三步了:

3. 依次执行下面命令：

python3 configure.py --cuda-root=/usr/local/cuda-11.4 
# 这里cuda-11.4 是你的cuda 文件夹名称，需要按照你的版本修改
make -j4  
sudo python3 setup.py install  

安装完成后，可以运行pycuda 提供的示例进行测试：

cd  examples
python3 demo.py

成功！