Spacy CLI 训练无法激活 GPU

Spacy CLI Training Unable to Activate GPU

我正在尝试在 CLI 上训练 NER Spacy 模型。按照所有必要的步骤,我最终创建了一个正确的输入文件,但是当我尝试在 GPU 上训练时,我收到消息说 spacy 无法激活 GPU,其他程序实际上可以使用我的 GPU 并且 cuda 设置正确。它似乎仍然不起作用,我的计算机中只有 1 个 GPU,所以我在 CLI 上选择了 -g 0。我找不到任何进一步的信息来说明为什么 GPU 无法激活,在互联网上搜索也没有任何结果。

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

Image showing the nvidia-smi output

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