问题情景
使用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
进行查看。
安装完成之后,须重启电脑,再次运行,成功转换。