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

【问题标题】：is there any way to train spacy on google colab gpu?有没有办法在 google colab gpu 上训练 spacy？
【发布时间】：2019-10-26 20:22:14
【问题描述】：

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

【问题讨论】：

【解决方案1】：

是的，有可能。

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

脚本：

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

GPU: True

【讨论】：

Google Colab 现在有 Cuda 10.1，每个“！nvcc --version”。然后你应该做“！pip install -U spacy[cuda101]”