RuntimeError: GET was unable to find an engine to execute this computation解决记录

发表于 2023-04-23 更新于 2023-09-25 本文字数： 127 阅读时长 ≈ 1 分钟

想尝试使用yolov5的源码跑数据的时候，环境配置好后，在测试是否能够跑数据的时候出现这个bug

出现这个问题，主要是cuda版本和torch版本不匹配导致的，解决方法就是看pytorch官方给的历史版本里找到对应的cuda版本的那个安装指令，然后安装torch。

参考资料：