Spacy CLI Training Не удалось активировать графический процессор - PullRequest
0 голосов
/ 29 апреля 2020

Я пытаюсь обучить модели NER Spacy в CLI. Выполнив все необходимые шаги, я, наконец, создал правильный входной файл, однако, пытаясь тренироваться на GPU, я получаю сообщение о том, что spacy не может активировать GPU, другие программы фактически могут использовать мой GPU, и cuda настроена правильно. Тем не менее, это не похоже на работу, у меня только 1 графический процессор на моем компьютере, поэтому я выбрал -g 0 в CLI. Я не могу найти дополнительную информацию о том, почему нельзя активировать графический процессор, поиск по inte rnet также ни к чему не привел.

Training pipeline: ['ner']
⚠ Unable to activate GPU: 0
Using CPU only
  • Версия драйвера NVIDIA: 440.64
  • CUDA-версия: 10,2
  • Графический процессор GeForce RTX 2060

Изображение, показывающее вывод NVIDIA-SMI

1 Ответ

0 голосов
/ 29 апреля 2020

Убедитесь, что у вас установлена ​​правильная версия cupy (из приведенной выше версии CUDA: cupy-cuda102).

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...