spacy in colab gets all-NaN slices

[diegorv1994]

I am able to train a spacy model on google Colab on None mode, but when I try to use Colab's GPU I am not able to train the model. All slices are equal to Nan. Does anyone know how to fix it?

y

[info2000]

did you installed the cuda113 package? and set prefer_gpu?
https://spacy.io/usage#gpu
also, take care not always colab enable a gpu

[diegorv1994]

Yes, I wrote require_gpu(). The output was True.

[polm]

I haven't seen that particular issue before, but take a look at the FAQ - colab seems to come with some special versions of libraries to support their hardware, and it seems most people are successful not installing cuda/cupy packages directly.

Also please include the output of spacy info and other information about the packages in your environment, such as the output of pip freeze.