"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"
有些早期task的pipeline可能不支持 cuda:1 这种写法,尝试在运行命令行的时候指定环境变量。 比如 CUDA_VISIBLE_DEVICES=0 python xxx 此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”
ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!欢迎加入技术交流群:微信公众号:魔搭ModelScope社区,钉钉群号:44837352