可以用m1芯片运行吗？
AriaZiyanYang opened this issue · 3 comments
AriaZiyanYang commented
作者您好，看到readme中给出的环境是基于cuda的，我的电脑是苹果m1芯片，目前pytorch还不支持m1的cuda版本，只有default版本，请问有什么解决办法吗？
AriaZiyanYang commented
作者您好，我搜了一些资料，发现m1芯片需要使用mps,但是您的trainer脚本中使用的是cuda,请问我是不是应该在trainer脚本中修改代码，来适应我的m1芯片？谢谢！
imlixinyang commented
这个你需要参考一下pytorch官方的教程或者示例了 我也不是很了解 不过训练的话一般还是用gpu好一些
AriaZiyanYang commented
好的，谢谢～