edgeyolo
edgeyolo copied to clipboard
AMd显卡,yolov5训练正常,edgeyolo提示错误
错误提示:
assert torch.cuda.is_available(), "cuda not available, please double check your device status!" AssertionError: cuda not available, please double check your device status!
我安装的是only cpu pytorch
如果使用cpu训练的话,需要自行改代码
好的,谢谢