Spacy CLI 训练无法激活 GPU

Question

我正在尝试在 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

• NVIDIA 驱动程序版本：440.64
• CUDA 版本：10.2
• GPU GeForce RTX 2060

Image showing the nvidia-smi output

Answer 1

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