用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赞同 展开评论 打赏
相关问答
热门讨论
热门文章
modelscope上跑报错，提示要pip install ttsfrd，搞不定
7549
关于 Windows平台上 ttsfrd 库的问题。
1540
KeyError: 'asr-inference is not in the pipelines r
1911
linux系统运行TTS语音合成报错 Could not initialize NNPACK
664
PyTorch Transformer 英中翻译超详细教程
877
目录空间都删的只剩2G了，还报OSError: [Errno 122] Disk quota exc
961
ttsfrd 是不开源的吗
3850
No module named 'ttsfrd'
1326
个性化语音合成训练 [ONNXRuntimeError]:7:INVALID_PROTOBUF
1057
报了很多警告，请问怎么解决？
982
展开全部
