ubuntu18.04安装tensorRT8.2实操

一、安装pycuda

按照参考文档几个命令安装，但会有个问题。如果是容器中安装注意要映射nvcc，最好把整个路径映射过来/usr/local/bin；

然后可能还会出现src/cpp/cuda.hpp:14:10: fatal error: cuda.h: No such file or directory这个问题，这个是在docker中可能会出现，要注意lib和include的位置变了，我的是在/usr/local/cuda-11.4/targets/x86_64-linux/，然后这么安装：

pip install --global-option=build_ext --global-option="-I/usr/local/cuda-11.4/targets/x86_64-linux/include/" --global-option="-L/usr/local/cuda-11.4/targets/x86_64-linux/lib/" -i https://pypi.tuna.tsinghua.edu.cn/simple  pycuda

二、安装TensorRT

这一步基本没什么问题，注意安装版本即可。

下载网址：https://developer.nvidia.com/tensorrt-getting-started

参考文档：

1. https://blog.csdn.net/laizi_laizi/article/details/121567609
2. https://stackoverflow.com/questions/52195608/src-cpp-cuda-hpp1410-fatal-error-cuda-h-no-such-file-or-directory