funasr: 报错 CUDA error: invalid device ordinal

问题描述

使用案例中的代码加载模型的时候，会报错 CUDA error: invalid device ordinal

运行的代码是

model = AutoModel(
    model=model_dir,
    vad_model="fsmn-vad",
    vad_kwargs={"max_single_segment_time": 30000},
    device="cuda:0",
)

解决问题

我得原因是 我再运行上面这段代码之前，运行了下面的代码：

os.environ['CUDA_VISIBLE_DEVICES'] = "0"

把这一行去掉了，就能正常运行了。

##

祝你好运

有问题可以交流呀

concat("求求QUN", "61480", "9646")