ModelScope进行推理的时候,进程还是使用第0块显卡, device参数没有生效,怎么解决?

"model_id = 'speech_tts/speech_sambert-hifigan_tts_zh-cn_multisp_pretrain_24k';sambert_hifigan_tts = pipeline(task=Tasks.text_to_speech, model=model_id, device='cuda:1')
使用上面的代码,ModelScope进行推理的时候,进程还是使用第0块显卡, device参数没有生效,怎么解决? 环境是 python3.8.19 modelscope==1.13.2"

展开
收起
小小爱吃香菜 2024-07-17 08:18:30 74 发布于吉林 分享
分享
版权
举报
1 条回答
写回答
取消 提交回答
  • 有些早期task的pipeline可能不支持 cuda:1 这种写法,尝试在运行命令行的时候指定环境变量。 比如 CUDA_VISIBLE_DEVICES=0 python xxx 此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2024-07-17 13:02:51 举报
    赞同 3 评论

    评论

    全部评论 (0)

    登录后可评论

ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!欢迎加入技术交流群:微信公众号:魔搭ModelScope社区,钉钉群号:44837352

热门讨论

热门文章

还有其他疑问?
咨询AI助理