Модель в keras не тренируется при использовании GPU, но работает на GPU на kaggle

Я тренирую модель с помощью model.fit_generator()

class batch_generator(Sequence):
    def __init__(self, type_):
        self.type = type_
    def __len__(self) :
        return 500
    def __getitem__(self, idx):
        data = x_train[idx*100:idx*100+100]
        labels = y_train[idx*100:idx*100+100]
        return data, labels

train_generator = batch_generator()

model.fit_generator(train_generator, use_multiprocessing = True)

x_train и y_train - MNIST из keras.dataset, model - маленькая сверточная нейронка, которая тренируется даже на СPU с нормальной скоростью
Что важно - если use_multiprocessing = False, то модель тренируется нормально
Если запустить этот же код в ноутбуке на kaggle, то модель работает нормально
Tensorflow только что обновил, видеокарта NVIDIA GEFORCE GTX 950M