Когда я выполняю nlp.begin_training (), я получаю следующую ошибку - PullRequest
Новая
       12

Когда я выполняю nlp.begin_training (), я получаю следующую ошибку

0 голосов
/ 05 июня 2019

"Я пытаюсь обучить модель с помощью spacy. Она прекрасно работает для CPU. Но когда я пытаюсь выполнить ее с помощью GPU, я получаю следующую ошибку. Spacy версия 2.1.4, CUDA версия 10.1"

«Я пытался переустановить thinc, но все равно получаю сообщение об ошибке» 

from __future__ import unicode_literals, print_function
import plac
import random
from pathlib import Path
import spacy
from spacy.util import minibatch, compounding
import json
spacy.require_gpu()
nlp = spacy.blank("en")
ner = nlp.create_pipe("ner")
ner = nlp.create_pipe("ner")
for _, annotations in TRAIN_DATA:
    for ent in annotations.get("entities"):
        ner.add_label(ent[2])
optimizer = nlp.begin_training()

«Я получаю следующую ошибку»

"CUDARuntimeError
Traceback (последний звонок последний) в ----> 1 optimizer = nlp.begin_training()

G: \ Anaconda3 \ lib \ site-packages \ spacy \ language.py в begin_training(self, get_gold_tuples, sgd, component_cfg, **cfg)

    547             if self.vocab.vectors.data.shape[1] >= 1:
    548                 self.vocab.vectors.data = Model.ops.asarray(self.vocab.vectors.data)
--> 549         link_vectors_to_models(self.vocab)
    550         if self.vocab.vectors.data.shape[1]:
    551             cfg["pretrained_vectors"] = self.vocab.vectors.name

G: \ Anaconda3 \ lib \ site-packages \ spacy_ml.py в link_vectors_to_models(vocab)

    297         else:
    298             word.rank = 0
--> 299     data = ops.asarray(vectors.data)
    300     # Set an entry here, so that vectors are accessed by StaticVectors
    301     # (unideal, I know)

ops.pyx in thinc.neural.ops.CupyOps.asarray()

G: \ Anaconda3 \ lib \ site-packages \ cupy \ creation \ from_data.py в array(obj, dtype, copy, order, subok, ndmin)

     39 
     40     """
---> 41     return core.array(obj, dtype, copy, order, subok, ndmin)
     42 
     43

cupy \ core \ core.pyx в cupy.core.core.array()

cupy \ core \ core.pyx в cupy.core.core.array()

cupy \ core \ core.pyx в cupy.core.core.ndarray.__init__()

cupy \ cuda \ memory.pyx в cupy.cuda.memory.alloc()

cupy \ cuda \ memory.pyx в cupy.cuda.memory.MemoryPool.malloc()

cupy \ cuda \ memory.pyx в cupy.cuda.memory.MemoryPool.malloc()

cupy \ cuda \ device.pyx в cupy.cuda.device.get_device_id()

cupy \ cuda \ runtime.pyx в cupy.cuda.runtime.getDevice()

cupy \ cuda \ runtime.pyx в cupy.cuda.runtime.check_status()

CUDARuntimeError: cudaErrorUnknown: неизвестная ошибка "

...