有什么方法可以在 google colab gpu 上训练 spacy 吗？

Question

我想在自定义数据集上训练 spacy 模型，但训练时间太长，有什么方法可以加快训练速度。我在 ner.begin_training() 中传递了 device=0 但它花费的时间与以前相同。

Answer 1

对的，这是可能的。

1. 转到[EDIT]-> [Notebook settings]-> 选择GPU （Hardware Acceleration它会重新启动运行时，因此所有单元格状态都会丢失）
2. 用于!pip install -U spacy[cuda100]通过 Cuda 支持安装 spacy
3. 运行以下命令

脚本：

import spacy
gpu = spacy.prefer_gpu()
print('GPU:', gpu)

它返回：

GPU: True