Open HildaM opened 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
由于仓库没有代码,不能提供pr了,希望作者大大采纳一下,照顾一下苹果用户
为pytorch增加M芯片的加速,配置很简单,如以下代码所示:
参考文档:https://pytorch.org/docs/stable/notes/mps.html