配置TensorRT
1、下载TensroRT:链接:https://developer.nvidia.com/nvidia-tensorrt-8x-download
2、将下载的文件进行解压,并将TensorRT中的include文件移动到CUDA文件夹下的include中,并将TensortRT文件中的lib中的dll与lib文件分别放到cuda中的bin与lib\x64文件夹中
3、安装TensorRT,分别对这四个文件进行安装
pip install tensorrt-8.2.5.1-cp37-none-win_amd64.whl
4、下载cudnn64_8.dll,并将该文件放到
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin 目录下
5、运行代码
import torch
import tensorrt as trt
print(trt.__version__)
print(torch.cuda.is_available())
6、运行示例,修改指定地方
7、下载cudnn:https://developer.nvidia.com/rdp/cudnn-download,将三个文件夹下的文件复制到NVIDIA GPU Computing Toolkit下对应的文件夹下
8、下载zlibwapi.dll文件,并将文件放在system32文件夹下,运行结果
若有误,请指正(谢谢)