Google colaboratory 有没有办法在google colab gpu上训练spacy？

我想在自定义数据集上训练spacy模型，但训练时间太长，有没有办法加快训练速度。 我在ner.begin_training（）中通过了device=0，但所花的时间与以前相同。

是的，这是可能的

转到

[编辑]

->

[笔记本设置]

->在

硬件加速下选择GPU（它会重新启动运行时，因此您的所有单元状态都会丢失）

使用

！pip安装-U spacy[cuda100]

使用Cuda支持安装spacy

运行以下命令

脚本：

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

它返回：

GPU: True

---

谷歌Colab现在每“！nvcc--version”拥有Cuda 10.1。然后您应该执行“！pip安装-U spacy[cuda101]”