參考 https://blog.csdn.net/zong596568821xp/article/details/86077553
1. 上網搜索安裝NVIDIA-DRIVER，CUDA10.1，CUDNN7.6.5 的方法，並安裝好。
2. 安裝pycuda：
如果要使用python接口的tensorrt，則需要安裝pycuda，命令：pip install 'pycuda>=2017.1.1'；
3. 下載TensorRT：
網址：https://developer.nvidia.com/nvidia-tensorrt-download；
注意：CUDA10.1對應TensorRT6；完整安裝，建議選擇Tar File Install Packages，這樣可以自行選擇安裝位置；
4. 安裝指令：
注意：如果使用anaconda建立虛擬環境×××，使用conda activate ×××；
#假設，在home下新建文件夾，命名為tensorrt_tar，然后將下載的壓縮文件拷貝進來解壓
tar xzvf TensorRT-6.××××.tar
#解壓得到TensorRT-6.××××的文件夾，將里邊的lib絕對路徑添加到環境變量中
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/用戶名/tensorrt_tar/TensorRT-6.××××/lib
#安裝TensorRT
cd TensorRT-6.××/python
pip install tensorrt-6.××.whl
注意：使用ls看一下目錄下文件，如果你是python3.7環境，對應選擇cp37版本，下面也一樣，如果只有單一文件，那就選那個；
#安裝UFF
cd TensorRT-6.××/uff
pip install uff-××××.whl
#安裝graphsurgeon
cd TensorRT-6.××/graphsurgeon
pip install graphsurgeon-××××.whl
5. 測試
import tensorrt
from tensorflow.python.framework import graph_util
如果沒有報錯，就代表安裝成功了；
6. 如果pycharm報錯缺少libnvinfer.so.6等：
ImportError: libnvinfer.so.6: cannot open shared object file: No such file or directory......
方法一：檢查 sudo gedit ~/.bashrc 中是否添加了TensorRT的環境變量，如果沒有，就將lib和bin添加到環境變量；
方法二：在pycharm的Run/Debug Configurations中添加環境變量 Environment variables：
注意：我這里的目錄和上文不一樣，你也要選擇你的TensorRT解壓目錄！
完畢。