torch=1.2时运行yolov5遇到的问题

Question

使用yolov5源码的时候需要torch=1.7这个代码才能运行

但是我的torch=1.2，请问怎么把这个代码调整成torch=1.2也能运行啊

Answer 1

torch.cuda.amp这个库和你的程序不兼容。检查下。