Как я могу запустить модель, обученную на GPU в spacy - PullRequest
Новая
       4

Как я могу запустить модель, обученную на GPU в spacy

1 голос
/ 01 марта 2020

Title.I использует spacy2.2.3 для обучения новой модели, поэтому я пишу код в cmd в windows7: 

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, и я узнал об этом с помощью справки. А потом я открыл taskmgr и GPU-Z, результат сказал мне, что он все еще работает на процессоре вместо gpu.

Почему это так? Я довольно нов в этом. Кто может мне помочь?

Система: Windows 7

Графический процессор: GT740

1 Ответ

0 голосов
/ 02 марта 2020

Две вещи:

  • -g -1 отключает графический процессор, опция включения графического процессора - -g 0 или -g 1 или какой идентификатор графического процессора вы хотите использовать

  • есть серьезная ошибка в поддержке графического процессора для windows в spacy v2.2.3, поэтому вы хотите тренироваться на CPU только в windows (пропустите -g или используйте -g -1 как указано выше), пока не выйдет v2.2.4

...