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赞同 171 展开评论 打赏
相关问答
热门讨论
热门文章
modelscope-funasr的FunClip怎么安装的？
1936
小白求助，跟着大模型微调教程做不太明白
797
练的pth模型文件，怎么转成bin文件？
905
在modelscope-funasr中找不到有效的 OpenPGP 数据，如何解决？
113
modelscope-funasr在安卓平台可以集成sense voice吗？
163
gpt3,加载数据时，MsDataset.load报错
1183
报错AssertionError: choose a window size 400 that is
250
基于SQL数据库的RAG的若干疑问
22
请问微调开元模型qwe1.5b 和 7b 分别支持数据集的上下文长度是多少个汉字/token？
1162
如何下载llama模型到本地？
2828
展开全部
智胜未来：国内大模型+Agent应用案例精选，以及主流Agent框架开源项目推荐
1720
今日 AI 开源｜共 4 项｜DeepSeek 推出新一代 AI 推理模型，实力比肩 OpenAI o1-preview！
34
RAG+AI工作流+Agent：LLM框架该如何选择，全面对比MaxKB、Dify、FastGPT、RagFlow、Anything-LLM,以及更多推荐
655
Perplexica：开源 AI 搜索引擎，Perplexity AI 的开源替代品，支持多种搜索模式、实时信息更新
25
ModelScope X 昇腾910快速上手
559
LLM 大模型学习必知必会系列(二)：提示词工程-Prompt Engineering 以及实战闯关
926
LLM 大模型学习必知必会系列(一)：大模型基础知识篇
1354
Memoripy：支持 AI 应用上下文感知的记忆管理 Python 库
21
告别Hugging Face模型下载难题：掌握高效下载策略，畅享无缝开发体验
391
2024年6月118篇代码大模型论文最全整理
703
展开全部
