SDT icon indicating copy to clipboard operation
SDT copied to clipboard
verified publisher icon dailenson

后续是否会发布支持Apple M1芯片的程式？

Open ENIGMA-K opened this issue 2 years ago • 2 comments

由于Apple M1芯片的MacBook不再能支持torch.device("cuda")，是否后续会发布基于M1芯片GPU训练的程式，如利用torch.device("mps")？或者现阶段有什么其他的解决办法？

https://blog.csdn.net/wangzaiyouzr/article/details/130140579 【MacOS】MacBook使用本机m1芯片GPU训练的方法（mps替代cuda）

ENIGMA-K avatar Dec 15 '23 14:12 ENIGMA-K

我查了一下，在M1芯片的GPU上训练似乎只要将源代码中的torch.device("cuda")替换成torch.device("mps")就好？

dailenson avatar Dec 19 '23 07:12 dailenson

并非所有的地方都能简单通过类似您说的方式解决，并且我也尝试过但没有完全解决

ENIGMA-K avatar May 25 '24 15:05 ENIGMA-K