search

dailenson / SDT

This repository is the official implementation of Disentangling Writer and Character Styles for Handwriting Generation (CVPR 2023)
MIT License
968 stars 82 forks source link

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

Open ENIGMA-K opened 9 months ago

ENIGMA-K commented 9 months ago

由于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）

dailenson commented 8 months ago

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

ENIGMA-K commented 3 months ago

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