I built the environment according to doc, and after run python backend/main.py got this error
Traceback (most recent call last):
File "backend/main.py", line 275, in <module>
sg.run()
File "backend/main.py", line 214, in run
recognizer = AudioRecogniser(language=self.language)
File "backend/main.py", line 32, in __init__
self.model = whisper.load_model(self.model_path)
File "/home/ddl/video_cut/video-subtitle-generator/backend/whisper/__init__.py", line 149, in load_model
model.load_state_dict(checkpoint["model_state_dict"])
File "/home/ddl/anaconda3/envs/vsgEnv/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1497, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for Whisper:
While copying the parameter named "encoder.blocks.0.attn.query.weight", whose dimensions in the model are torch.Size([1024, 1024]) and whose dimensions in the checkpoint are torch.Size([1024, 1024]), an exception occurred : ('CUDA error: no kernel image is available for execution on the device',).
While copying the parameter named "encoder.blocks.0.attn.key.weight", whose dimensions in the model are torch.Size([1024, 1024]) and whose dimensions in the checkpoint are torch.Size([1024, 1024]), an exception occurred : ('CUDA error: no kernel image is available for execution on the device',).
While copying the parameter named "encoder.blocks.0.attn.value.weight", whose dimensions in the model are torch.Size([1024, 1024]) and whose dimensions in the checkpoint are torch.Size([1024, 1024]), an exception occurred : ('CUDA error: no kernel image is available for execution on the device',).
While copying the parameter named "encoder.blocks.0.attn.out.weight", whose dimensions in the model are torch.Size([1024, 1024]) and whose dimensions in the checkpoint are torch.Size([1024, 1024]), an exception occurred : ('CUDA error: no kernel image is available for execution on the device',).
My GPU is 3090, what should I do about it? Looking forward to your reply!
I built the environment according to doc, and after run
python backend/main.py
got this errorMy GPU is 3090, what should I do about it? Looking forward to your reply!