可以用m1芯片运行吗？

imlixinyang / HiSD

Code for "Image-to-image Translation via Hierarchical Style Disentanglement" (CVPR 2021 Oral).

Other

391 stars 49 forks source link

Closed Ziyan-swu closed 2 years ago

Ziyan-swu commented 2 years ago

作者您好，看到readme中给出的环境是基于cuda的，我的电脑是苹果m1芯片，目前pytorch还不支持m1的cuda版本，只有default版本，请问有什么解决办法吗？

Ziyan-swu commented 2 years ago

作者您好，我搜了一些资料，发现m1芯片需要使用mps,但是您的trainer脚本中使用的是cuda,请问我是不是应该在trainer脚本中修改代码，来适应我的m1芯片？谢谢！

imlixinyang commented 2 years ago

这个你需要参考一下pytorch官方的教程或者示例了我也不是很了解不过训练的话一般还是用gpu好一些

Ziyan-swu commented 2 years ago

好的，谢谢～