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

Question

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

Answer 1

对的，这是可能的。

转至 [EDIT] -> [Notebook settings] -> 在 Hardware Acceleration 下选择 GPU (它会重新启动运行时，因此所有单元格状态都会丢失)

使用 !pip install -U spacy[cuda100]使用 Cuda Support 安装 spacy

运行以下

脚本:

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

它返回:

GPU: True