nvidia_deeplearningexamples_tacotron2: RuntimeError: ошибка CUDA: недопустимая функция устройства

Question

Настройка времени выполнения: python3 и GPU.

Шаг за шагом запускайте код . Я только успешно запускаю код в первый раз.

После этого при запуске следующей части произошла ошибка RuntimeError: CUDA: недопустимая функция устройства

sequence = np.array(tacotron2.text_to_sequence(text, ['english_cleaners']))[None, :]
sequence = torch.from_numpy(sequence).to(device='cuda', dtype=torch.int64)

with torch.no_grad():
    _, mel, _, _ = tacotron2.infer(sequence)
    audio = waveglow.infer(mel)
audio_numpy = audio[0].data.cpu().numpy()
rate = 22050

Знаете ли вы причину? И работает ли предварительно обученная модель на локальном процессоре?

Answer 1

На момент написания этой статьи вы можете решить эту проблему, добавив! Pip install torch == 1.1.0 torchvision == 0.3.0 до импорта torch в https://colab.research.google.com/github/pytorch/pytorch.github.io/blob/master/assets/hub/nvidia_deeplearningexamples_tacotron2.ipynb