解决办法：使用DataParalle包裹模型

model = nn.DataParallel(model)

 

Logo
AI编程社区

汇聚全球AI编程工具，助力开发者即刻编程。

更多推荐

cover

2025年上半年前端技术圈生态总结

avatar AI编程社区
cover

使用Llama3/Qwen2等开源大模型，部署团队私有化Code Copilot和使用教程，大模型入门到精通，收藏这篇就足够了！

avatar AI编程社区
cover

Kimi K2-0905重磅发布：月之暗面再次引领AI编程新纪元

avatar AI编程社区