modelscope-funasr怎么设置使得模型可以用CUDA0以外的其他gpu？
modelscope-funasr怎么设置使得模型可以用CUDA0以外的其他gpu？设置device='cuda:1'模型在cuda 0之外的gpu上面做推理就会出现 RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cuda:0
展开
收起
1
条回答
写回答
写回答
-
北京阿里云ACE会长模型是基于PyTorch构建的，确保在模型加载和推理时，通过.to(device)方法将模型转移到目标GPU上，这里的device应该是你想要使用的GPU，比如device = torch.device('cuda:1')，然后model.to(device)。
确保所有输入数据、中间变量以及模型都在同一个GPU上。如果使用 DataLoader，可以通过num_workers参数指定使用哪个GPU，并且在数据加载时就将数据转移到目标GPU。
2024-09-02 09:20:33赞同 111 展开评论 打赏
相关问答
热门讨论
热门文章
modelscope-funasr的FunClip怎么安装的？
1486
小白求助，跟着大模型微调教程做不太明白
415
open webui的RAG功能根本用不起来
864
modelscope-funasr启动ASR服务的时候一直在提示ssl证书错误是咋回事？
186
请问微调开元模型qwe1.5b 和 7b 分别支持数据集的上下文长度是多少个汉字/token？
1084
modelscope-funasr在安卓平台可以集成sense voice吗？
94
练的pth模型文件，怎么转成bin文件？
813
如何下载llama模型到本地？
2733
modelscope-funasr有没有开源的语音唤醒，自定义唤醒？
350
无法从tokenizer.model解析解析ModelProto
308
展开全部
智胜未来：国内大模型+Agent应用案例精选，以及主流Agent框架开源项目推荐
1146
2024年6月118篇代码大模型论文最全整理
622
LLM 大模型学习必知必会系列(二)：提示词工程-Prompt Engineering 以及实战闯关
874
【AI 技术分享】大模型与数据检索的探索实践
126
前端大模型应用笔记（二）：最新llama3.2小参数版本1B的古董机测试 - 支持128K上下文，表现优异，和移动端更配
48
ModelScope X 昇腾910快速上手
501
RAG：AI大模型联合向量数据库和 Llama-index，助力检索增强生成技术
590
检索增强生成(RAG)实践：基于LlamaIndex和Qwen1.5搭建智能问答系统
1024
LLM 大模型学习必知必会系列(一)：大模型基础知识篇
1272
DashVector&千问7B模型，打造极简RAG 之 基于专属知识的问答服务实践
1018
展开全部
