基于directml的yolov5运行问题

当device=‘dml'时报错,然后我把model = torch.jit.load(w) if 'torchscript' in w else attempt_load(weights, map_location=device)改成model = torch.jit.load(w) if 'torchscript' in w else attempt_load(weights, map_location='cpu').to(device)时它不报错,但是图片识别不出来,device=cpu时可以正常识别,有啥解决办法吗显卡amdrx550

这个你估计得去微软的directml底下去问了,还是那句话,深度学习请用N卡,各大框架官方的支持都是cuda

哥们,你这个问题解决了么?

可能你的是笔记本,建议将python.exe在windows设置里面的图形设置,浏览找到python.exe设置成高性能模式(就是对应你电脑的独立显卡)就可以了