edgeyolo icon indicating copy to clipboard operation
edgeyolo copied to clipboard

AMd显卡,yolov5训练正常,edgeyolo提示错误

Open g101400 opened this issue 2 years ago • 3 comments

错误提示:

assert torch.cuda.is_available(), "cuda not available, please double check your device status!" AssertionError: cuda not available, please double check your device status!

g101400 avatar Feb 21 '23 09:02 g101400

我安装的是only cpu pytorch

g101400 avatar Feb 21 '23 09:02 g101400

如果使用cpu训练的话,需要自行改代码

LSH9832 avatar Feb 21 '23 10:02 LSH9832

好的,谢谢

g101400 avatar Feb 22 '23 10:02 g101400