TRT使用之pycuda安装

执行：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

如果有找不到 pyconfig.h 的错误，就安装 python3-dev。

如果有找不到 nvcc 的错误，就 export CUDA_ROOT=/usr/local/cuda/bin。

参考：