amd显卡yolov5怎么使用directml推理

有没有大佬解答一下呀？我只有python版本的yolov3的directml推理源码，我看了一下，两个地方没有什么大区别只是torch_utils.py这里有点不太一样其余的都是在模块那里加上import directml有没有大佬可以帮下忙呀？如果需要我可以提供V3的源码

这个玩意只能你自己一点一点的修改，咱也没试过，只知道这个可以支持windows下的A卡。目前手上没有A卡，深度学习还是cuda吧。
下载个代码对比的软件，然后对比下yolov3的源码和你能跑dml的源码，看下修改了哪些，然后yolov5对应的修改，最后运行，有报错一点一点的改吧。