有没有办法在 Google Colab GPU 上训练 Spacy？

Question

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

- vkypn

Answer 1

是的，这是可能的。

1. 进入 [编辑] -> [笔记本设置] -> 在 硬件加速 下选择 GPU（运行时将重新启动，因此所有单元格状态都会丢失）
2. 使用 !pip install -U spacy[cuda100] 安装支持Cuda的spacy
3. 运行以下脚本

脚本：

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

它返回：

GPU: True

- seb

---

Google Colab现在有Cuda 10.1，通过“！nvcc --version”可查看。然后您应该执行“！pip install -U spacy[cuda101]”。 - Carlos Aguayo