imlixinyang / HiSD

Official pytorch implementation of paper "Image-to-image Translation via Hierarchical Style Disentanglement" (CVPR 2021 Oral).

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

可以用m1芯片运行吗?

Ziyan-swu opened this issue · comments

commented

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

commented

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

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

commented

好的,谢谢~