Spacy培训无法激活GPU

Question

我正在尝试在CLI上训练一个NER Spacy模型。按照所有必要的步骤，我最终创建了一个正确的输入文件，但是当我试图在GPU上进行训练时，我得到了spacy无法激活GPU的消息，其他程序实际上能够使用我的GPU，并且cuda被正确设置。尽管如此，它似乎不工作，我只有一个GPU在我的电脑，所以我选择了-g 0的CLI。我找不到任何进一步的信息，因为为什么GPU不能被激活，搜索互联网也没有任何结果。

Training pipeline: ['ner']
⚠ Unable to activate GPU: 0
Using CPU only

• 英伟达-驱动程序-版本: 440.64
• 库达-verion: 10.2
• GPU GeForce RTX 2060

显示nvidia-smi输出的图像。

Answer 1

检查您是否安装了正确的cupy版本(来自上面的CUDA版本：cupy-cuda102)。