语音识别阶段出错[阿里FunASR(本地)(本地内置)] CUDA out of memory. Tried to allocate 912.00 MiB. GPU 0 has a total capacity of 4.00 GiB of which 608.65 MiB is free. Of the allocated memory 1.65 GiB is allocated by PyTorch, and 993.45 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to avoid fragmentation. See documentation for Memory Management (https://pytorch.org/docs/stable/notes/cuda.html#environment-variables):Traceback (most recent call last):
File "videotrans\process\stt_fun.py", line 490, in paraformer
File "D:\download\pyVideoTrans\_internal\modelscope\pipelines\audio\funasr_pipeline.py", line 73, in call
output = self.model(*args, **kwargs)File "modelscope\models\base\base_model.py", line 37, in call
return self.postprocess(self.forward(*args, **kwargs))File "D:\download\pyVideoTrans\_internal\modelscope\models\audio\funasr\model.py", line 61, in forward
ou......
, in run
File "videotrans\recognition\_base.py", line 90, in run
File "videotrans\recognition\_funasr.py", line 61, in _exec
File "videotrans\configure\base.py", line 272, in _new_process
videotrans.configure.excepts.VideoTransError: CUDA out of memory. Tried to allocate 912.00 MiB. GPU 0 has a total capacity of 4.00 GiB of which 608.65 MiB is free. Of the allocated memory 1.65 GiB is allocated by PyTorch, and 993.45 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to avoid fragmentation. See documentation for Memory Management (https://pytorch.org/docs/stable/notes/cuda.html#environment-variables):Traceback (most recent call last):
File "videotrans\process\stt_fun.py", line 490, in paraformer
File "D:\download\pyVideoTrans\_internal\modelscope\pipelines\audio\funasr_pipeline.py", line 73, in call
output = self.model(*args, **kwargs)File "modelscope\models\base\base_model.py", line 37, in call
return self.postprocess(self.forward(*args, **kwargs))File "D:\download\pyVideoTrans\_internal\modelscope\models\audio\funasr\model.py", line 61, in forward
output = self.model.generate(*args, **kwargs)File "funasr\auto\auto_model.py", line 329, in generate
return self.inference_with_vad(File "funasr\auto\auto_model.py", line 506, in inference_with_vad
results = self.inference(File "funasr\auto\auto_model.py", line 381, in inference
res = model.inference(**batch, **kwargs)File "D:\download\pyVideoTrans\_internal\funasr\models\seaco_paraformer\model.py", line 429, in inference
_, _, us_alphas, us_peaks = self.calc_predictor_timestamp(File "D:\download\pyVideoTrans\_internal\funasr\models\bicif_paraformer\model.py", line 139, in calc_predictor_timestamp
ds_alphas, ds_cif_peak, us_alphas, us_peaks = self.predictor.get_upsample_timestamp(File "D:\download\pyVideoTrans\_internal\funasr\models\bicif_paraformer\cif_predictor.py", line 265, in get_upsample_timestamp
output2, (_, _) = self.blstm(output2)File "torch\nn\modules\module.py", line 1751, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)File "torch\nn\modules\module.py", line 1762, in _call_impl
return forward_call(*args, **kwargs)File "torch\nn\modules\rnn.py", line 1124, in forward
result = _VF.lstm(torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 912.00 MiB. GPU 0 has a total capacity of 4.00 GiB of which 608.65 MiB is free. Of the allocated memory 1.65 GiB is allocated by PyTorch, and 993.45 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to avoid fragmentation. See documentation for Memory Management (https://pytorch.org/docs/stable/notes/cuda.html#environment-variables)
cfg=TaskCfgSTT(uuid='50d0170e4c', name='D:/张建光/姜胡说/2022年/@姜胡说_20220516_看到_打断_重新设计_执行路径_这很高级_姜胡....mp4', dirname='D:/张建光/姜胡说/2022年', noextname='@姜胡说_20220516_看到_打断_重新设计_执行路径_这很高级_姜胡...', basename='@姜胡说_20220516_看到_打断_重新设计_执行路径_这很高级_姜胡....mp4', ext='mp4', target_dir='D:/download/pyVideoTrans/output/recogn', cache_folder='D:/download/pyVideoTrans/tmp/12240/50d0170e4c', is_cuda=True, source_language=None, source_language_code=None, source_sub=None, source_wav=None, source_wav_output=None, target_language=None, target_language_code=None, target_sub='D:/download/pyVideoTrans/output/recogn/@姜胡说_20220516_看到_打断_重新设计_执行路径_这很高级_姜胡....srt', target_wav=None, target_wav_output=None, detect_language='zh-cn', recogn_type=3, model_name='paraformer-zh', shibie_audio='D:/download/pyVideoTrans/tmp/12240/50d0170e4c/@姜胡说_20220516_看到_打断_重新设计_执行路径_这很高级_姜胡...-1782441178.926296.wav', remove_noise=False, enable_diariz=False, nums_diariz=0, rephrase=0, fix_punc=0)
=
system:Windows-10-10.0.26200-SP0
version:v4.03
frozen:True
language:zh
root_dir:D:/download/pyVideoTrans