Stable Diffusion 可不可以应用在分布式的集群上?

2023-09-12 19:42:43 +08:00
 hanguofu
有没有人做过类似的研究/尝试?对硬件设备有没有什么限制?
1386 次点击
所在节点    奇思妙想
2 条回复
chesha1
2023-09-13 11:05:11 +08:00
1. 当然可以啊,Stable Diffusion 的原公司就是在一个 32 台 8 卡 A100 的集群上进行的训练

2. 不过这个方面的公开资料很少,因为大规模 GPU 集群是很工业界的东西,发论文的人不多,不如算法论文多
我也不是很了解这些,比如 google 的 pathway 就是相关工作,还有 OSDI/SOSP 上也有一些 MLsys 的论文
如果你的意思是怎么使用,而不是怎么研究,我也不清楚,我财力有限,没用过多台机器

3. 硬件限制没太懂,你是指什么样的限制呢?只要是服务器上有 GPU ,服务器之间用以太网/IB 连接,都能做,只是用不同的技术可能效率会差很多
RVRGnXxwawP5kwKW
2023-09-13 15:28:36 +08:00
大型 cnn 的分布式训练和部署已经很成熟了,这方面的论文每年 OSDI 都会有。工业界的方案早的有 mu li 的 parameter server ,近的有 Google 的 pathway 。

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/973125

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX