Cuda error: no kernel image is available for execution on the device #31285

huggingface / transfer-learning-conv-ai

🦄 State-of-the-Art Conversational AI with Transfer Learning

MIT License

1.73k stars 431 forks source link

Cuda error: no kernel image is available for execution on the device #31285 #101

Closed CiaraG98 closed 3 years ago

CiaraG98 commented 3 years ago

I get this error when trying to fine-tune the model with train.py with my own dataset. Using cuda 11.2 and pytorch 1.7.1+cu110. When running with CPU the training seems to make my laptop restart with some error, so I am trying to run with GPU instead.

Is there something I am missing that is making this not work?

CiaraG98 commented 3 years ago

Resolved this by installing conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch pip install install torch==1.3.0 pytorch-ignite transformers==2.5.1 tensorboardX==1.8 tensorflow spacy