video-subtitle-generator icon indicating copy to clipboard operation
video-subtitle-generator copied to clipboard

CUDA error: no kernel image

Open Yead opened this issue 10 months ago • 1 comments

image

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!

Yead avatar Aug 28 '23 09:08 Yead