我如何在Spacy中运行在GPU上训练过的模型

时间:2020-03-01 06:07:45

标签: python spacy

标题。我使用spacy2.2.3训练新模型,所以我在Windows7的cmd中进行编码:

python -m spacy train -g -1 zh models train_path dev_path

另外,我安装cupy-cuda101,版本7.2.0。在python3.7.6中编码时:

import spacy
spacy.prefer_gpu()
spacy.require_gpu()

它们都打印“ True”。

我使用'-g -1'是因为我想在gpu上对其进行训练,并且我通过help选项了解了这一点。 然后,我同时打开taskmgr和GPU-Z,结果告诉我它仍然在cpu而不是gpu上运行。

是什么原因造成的?我对此很陌生。谁可以帮助我?

系统:Windows 7

GPU:GT740

1 个答案:

答案 0 :(得分:0)

两件事:

  • -g -1禁用GPU,启用GPU的选项为-g 0-g 1或您希望使用的GPU ID

  • spacy v2.2.3中对Windows的GPU支持存在一个主要错误,因此您只想在Windows中的CPU上进行训练(省去-g或像上面一样使用-g -1)直到发布v2.2.4