【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进行查看。

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

猜你喜欢

转载自blog.csdn.net/qq1198768105/article/details/130031509