huchang47 / AetherConverTools

以太流派的AI转绘工具包
210 stars 25 forks source link

Apple M芯片加速支持 #15

Open HildaM opened 11 months ago

HildaM commented 11 months ago

为pytorch增加M芯片的加速,配置很简单,如以下代码所示:

if torch.cuda.is_available():
    device = torch.device("cuda")
    print("加速成功!使用的设备:CUDA")
elif torch.backends.mps.is_available():
    device = torch.device("mps")
    print("Apple M芯片使用 mps 加速")
else:
    device = torch.device("cpu")
    print("加速失败!使用的设备:CPU")

参考文档:https://pytorch.org/docs/stable/notes/mps.html

HildaM commented 11 months ago

由于仓库没有代码,不能提供pr了,希望作者大大采纳一下,照顾一下苹果用户