Linux 35.6 + JetPack v5.1.4之 pyCUDA升级

1. 源由

其实问题来自这里：How to install CUDA 11.8 on Jetpack 5.1.4 L4T 35.6?

当11.8 CUDA升级完成后，发现有一个版本信息显示为11.4。就想着升级的时候应该都是11.8。

经过帖子的讨论，结果说要升级pyCUDA，而实际不是，应该是OpenCV的支持目前是11.4的，需要重新编译OpenCV。

若需要将这个现实正确显示未CUDA 11.8，那么请参考：Linux 35.6 + JetPack v5.1.4@python opencv安装

2. 步骤

这里就是编译和异常解决的问题，不再展开，仅给出实际操作步骤。

2.1 依赖环境

$ sudo apt-get install ctags
$ pip install pytools==2023.1

2.2 获取源代码

$ git clone git@github.com:inducer/pycuda.git
$ cd pycuda
$ git checkout tags/v2024.1.2
$ git checkout -b nvidia_v2024.1.2

2.3 获取子模块

$ git submodule init
$ git submodule update

2.4 配置编译

$ python configure.py --cuda-root=/usr/local/cuda/targets/aarch64-linux
$ export PATH=${PATH}:/usr/local/cuda/bin
$ export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda/lib64
$ export CPATH=$CPATH:/usr/local/cuda/targets/aarch64-linux/include
$ export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/targets/aarch64-linux/lib
$ export PATH=/usr/local/cuda/bin:$PATH
$ export CUDA_INC_DIR=/usr/local/cuda/include
$ sudo make install

2.5 安装pyCUDA

$ pip install .

3. 总结

验证下安装的版本，与checkout的版本号一致。

$ python -c "import pycuda; print(pycuda.VERSION)"
(2024, 1, 2)

• jetson-fpv pyCUDA 版本打印

4. 问题

python 3.8 配套 pytools==2023.1 如果不配套就会安装报错，要升级python 3.10.

Requirement already satisfied: pytools>=2011.2 in /usr/local/lib/python3.8/dist-packages/pytools-2025.1.1-py3.8.egg (from pycuda==2024.1.2) (2025.1.1)
ERROR: Package 'pytools' requires a different Python: 3.8.10 not in '>=3.10'