执行:pip install pycuda
出现:Building wheel for pycuda (PEP 517) … error
使用:pip install pycuda --no-binary :all:
不解决问题。
从源码编译
1、下载 pycuda-2021.1.tar.gz
2、tar xfz pycuda-2021.1.tar.gz
3、cd pycuda-2021.1
4、python3 configure.py --cuda-root=/usr/local/cuda
5、make install
出现以下问题,
首先查看自己的 gcc 和 g++版本是否一致,没安装 g++ 的 安装 g++
然后查看配置文件中的 BOOST_COMPILER
是否和自己的 gcc 版本一致,不一致改成一致的
如果有找不到 pyconfig.h 的错误,就安装 python3-dev。
如果有找不到 nvcc 的错误,就 export CUDA_ROOT=/usr/local/cuda/bin。
参考链接:
TRT使用之pycuda安装
pycuda源码安装报错gcc failed with exit status -1
安装pycuda及问题解决
RedHat 安装 g++
分割线(以下为无关内容,仅用于记录)
flask 启动 tensorrt 模型,推理报错
https://github.com/jkjung-avt/tensorrt_demos/issues/213