windows上安装Tensorrt（python）

准备工作

安装CUDA和CUDNN，我安装CUDA版本是11.0
终端输入 nvcc --version

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Thu_Jun_11_22:26:48_Pacific_Daylight_Time_2020
Cuda compilation tools, release 11.0, V11.0.194
Build cuda_11.0_bu.relgpu_drvr445TC445_37.28540450_0

tensorrt安装

1. 下载tensorrt，链接，注意要下载8.2版本以后的，8.2版本以前的tensorrt不支持在windows上的python语言，只支持C++语言。
2. 解压以后，我把文件放在了E盘，E:\TensorRT-8.2.1.8
3. 在终端，创建一个虚拟环境，命名为trt，python语言选择的是3.7
4. pip install E:\TensorRT-8.2.1.8\python\tensorrt-8.2.1.8-cp37-none-win_amd64.whl cp37代表语言为python3.7
5. 添加系统环境变量，
6. E:\TensorRT-8.2.1.8中include复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0中，
7. E:\TensorRT-8.2.1.8中lib中的dll文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin中
8. E:\TensorRT-8.2.1.8中lib中lib的文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\lib中
9. windows系统中的环境变量
   
10. 然后下载一个zlib包，解压缩后找到zlibwapi.dll文件，剪切到C:\Windows\System32位置下面（这是cudnn依赖的动态链接库）

http://www.winimage.com/zLibDll/zlib123dllx64.zip

安装pycuda

链接

激活虚拟环境
pip install c:\users\admin\downloads\pycuda-2021.1+cuda115-cp37-cp37-win_amd64.whl