首页
学习
活动
专区
圈层
工具
发布
社区首页 >专栏 >module must have parameters on cuda:0 but found cuda:2:DataParallel 主卡与模型位置要一致

module must have parameters on cuda:0 but found cuda:2:DataParallel 主卡与模型位置要一致

作者头像
用户12833544
发布于 2026-11-14 14:57:10
发布于 2026-11-14 14:57:10
10
举报
概述
原帖把模型搬到 cuda:2,却用 device_ids=[0,1,2,3] 包装 DataParallel;后者要求基础模型在列表第一张卡:

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档