Enable GPU for Spacy

Question

I'm trying to set up a GPU to train models using Spacy

I've configured a docker container with GPU and it seems like it can be seen from Pytorch

Following the spacy site, to enable GPU I run the command

pip install -U 'spacy[cuda-autodetect]' 

It fails saying it doesn't detect CUDA

If I run the PyTorch collect_env script this is what I get

If I don't try to install this library and run spacy.prefer_gpu I get a False

I'm not sure what could be missing

As a side note, I can use the GPU with other libraries like transformers

Answer 1

Try spacy.require_gpu() which provides information why gpu is not available. For example: