search

surajiyer / spacycaKE

Simple keyphrase extraction extensions and pipeline components for spaCy.
MIT License
4 stars 0 forks source link

CUDA error #3

Open rkromann opened 3 years ago

rkromann commented 3 years ago

Hi, I have tried to run the simple_keyphrase_extraction.ipynb notebook using a GPU. It works to some degree, but I think there is a problem with the memory handling. I get errors like:

RuntimeError: cuda runtime error (700) : an illegal memory access was encountered at C:/cb/pytorch_1000000000000/work/aten/src/THC/THCTensorSort.cu:62

and:

RuntimeError: CUDA error: an illegal memory access was encountered

I am running Python 3.8.5 and Pytorch 1.6