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")