用inference_pipeline时无法指定gpu
代码如下：
inference_pipeline = pipeline( task = Tasks.auto_speech_recognition, model = "damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch", device = "cuda:0", )
无论device怎么设置，都会自动把数据放到一块固定的GPU上
展开
收起
1
条回答
写回答
写回答
-
可以使用CUDA_VISIBLE_DEVICES参数指定使用那张卡，cpu解码的话，把CUDA_VISIBLE_DEVICES参数设置成-1 例子： GPU 比如想指定用第3卡跑解码 CUDA_VISIBLE_DEVICES=3 python infer.py CPU CUDA_VISIBLE_DEVICES=-1 python infer.py
2023-02-22 17:39:45赞同 展开评论 打赏
相关问答
热门讨论
热门文章
KeyError: 'asr-inference is not in the pipelines r
1384
KanttsTrainer: Please install kantts.
581
魔搭modelscope按官方文档安装音频环境失败
466
关于 Windows平台上 ttsfrd 库的问题。
517
在Windows平台上关于 ModelScorpe Audio 安装涉及 kwsbp 的问题
305
pip安装语音领域模型依赖报错。
258
Paraformer语音识别-中文-通用-16k-离线-large-长音频版，无网络时无法使用如何处
365
kantts 这个模型是什么东西啊，运行会报错
1675
SambertHifigan个性化语音合成-中文-预训练-16k 语速太快了怎么设置慢一点
901
关于 modelscope-funasr 报错的问题。
860
展开全部
