今天在调试语音转文字工具FunASR的时候遇到了一个问题,开启语音识别后软件报错,部分输出信息如下:
File “D:\python\FunASR_20250208\funasr\auto\auto_model.py”, line 301, in generate return self.inference(input, input_len=input_len, **cfg) File “D:\python\FunASR_20250208\funasr\auto\auto_model.py”, line 372, in inference with torch.cuda.device(): TypeError: device.__init__() missing 1 required positional argument: ‘device’
报错提示torch.cuda.device()需要一个参数,这个参数是一个整数数值,也就是GPU的序号。
![](https://nuowa.net/wp-content/uploads/2025/02/image.png)
打开任务管理器在【性能】里可以看到电脑上的GPU数量和序号,比如说我电脑上现在只有一个GPU,序号就是0,我想使用这个GPU去处理的话,打开auto_model.py,这行报错代码可以改为:
with torch.cuda.device(0):
如果你有多个GPU,想用哪个处理这里就填哪个序号。