Linux 35.6 + JetPack v5.1.4之 pyCUDA升级
- 1. 源由
- 2. 步骤
- 2.1 依赖环境
- 2.2 获取源代码
- 2.3 获取子模块
- 2.4 配置编译
- 2.5 安装pyCUDA
- 3. 总结
- 4. 问题
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'
