SDT icon indicating copy to clipboard operation
SDT copied to clipboard

Published 20 hours ago • verified publisher icon dailenson

Mac M1 不支持Pytorch的CUDA. 因为木有英伟达显卡

Open MaySoon2020 opened this issue 1 year ago • 2 comments

MaySoon2020 avatar Dec 28 '23 19:12 MaySoon2020

找到全部.cuda() 修改为 to('cpu')

TxcA avatar Dec 29 '23 00:12 TxcA

在M1芯片的GPU上训练，试试将源代码中的torch.device("cuda")替换成torch.device("mps")？

dailenson avatar Dec 29 '23 03:12 dailenson