【Bug解决】TensorRT: export failure 0.1s: [WinError 127]

问题情景

使用YOLOv5将Pytorch(.pt)模型转换成TensorRT(.engine)格式时，遇到如下报错：

TensorRT: export failure 0.1s: WinError 127

解决过程

首先是根据我之前的博文【目标检测】使用TensorRT加速YOLOv5正确安装好TensorRT，目前使用的Cuda版本是11.8。

查阅官方仓库中的相关issue，主要解决思路是保证Cuda版本、Pytorch版本的一致性。

我的pytorch安装版本为1.13.0+cu117，因此换用11.7的Cuda版本。

各Cuda版本的下载链接可参考博文：NVIDIA CUDA各版本下载链接(包括最新12、11版本和以往10.2版本）

注，一台计算机上可以共存多个cuda版本，安装完成之后，修改相应环境变量即可。

版本验证可以在控制台中输入nvcc -V进行查看。

安装完成之后，须重启电脑，再次运行，成功转换。