有什么办法可以在Google Colab GPU上训练spacy？

Question

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

Answer 1

是的，有可能。

1. 转到[EDIT]-> [Notebook settings]->在Hardware Acceleration下选择 GPU （它将重新启动运行时，因此所有单元状态都将丢失）
2. 使用!pip install -U spacy[cuda100]安装带有Cuda支持的spacy
3. 运行以下内容

脚本：

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

它返回：

GPU: True