开发者社区 > ModelScope模型即服务 > 正文

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 26 0
1 条回答
写回答
取消 提交回答
  • 有些早期task的pipeline可能不支持 cuda:1 这种写法,尝试在运行命令行的时候指定环境变量。 比如 CUDA_VISIBLE_DEVICES=0 python xxx 此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2024-07-17 13:02:51
    赞同 3 展开评论 打赏

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

相关电子书

更多
视觉AI能力的开放现状及ModelScope实战 立即下载
ModelScope助力语音AI模型创新与应用 立即下载
服务上云加速大家居产业C2M进程 立即下载

相关实验场景

更多