用户12833544
module must have parameters on cuda:0 but found cuda:2:DataParallel 主卡与模型位置要一致
原创
关注作者
腾讯云
开发者社区
文档
建议反馈
控制台
登录/注册
首页
学习
活动
专区
圈层
工具
MCP广场
文章/答案/技术大牛
搜索
搜索
关闭
发布
用户12833544
社区首页
>
专栏
>
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
1
0
举报
概述
原帖把模型搬到 cuda:2,却用 device_ids=[0,1,2,3] 包装 DataParallel;后者要求基础模型在列表第一张卡:
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系
cloudcommunity@tencent.com
删除。
pytorch
问题归档
专栏文章
快讯文章归档
关键词归档
开发者手册归档
开发者手册 Section 归档