Jetson Xavier NX安装pycuda，报错解决方案

Jetson Xavier NX安装pycuda，报错解决方案

1. 安装必要的依赖项

sudo apt-get install libboost-all-dev
sudo apt-get install python-dev python-setuptoolslibboost-python-dev libboost-thread-dev -y

2. 下载软件包兵解压
首先选择与自己cuda对应版本的pycuda，本文已安装jetpack4.6，cuda10.2
到官方网址下载对应软件包pycuda-2021.1到/usr/local/目录下，并解压。

sudo tar xzvf pycuda-2021.1.tar.gz
cd /usr/local/pycuda-2021

3.安装

python3 configure.py --cuda-root=/usr/local/cuda-10.2
make -j4     # 解决各种的关键步骤，先编译
sudo python3 setup.py install

4. 验证
安装完成后，可以通过测试pycuda-2021.1里面的例子来验证安装：

cd /usr/local/pycuda-2021.1/examples
python3 demo.py

如果出现正确的矩阵计算结果，则表示PyCUDA安装成功。
PyCUDA安装路径为/usr/local/pycuda-2021.1

点赞支持一下吧～