Pytorch nvidia_deeplearningexamples_tacotron2:运行时错误：CUDA错误：无效的设备函数

设置运行时：python3和GPU

一步一步地跑。 我只在第一次成功运行代码

之后，当运行下面的部分时，出现“RuntimeError:CUDA error:invalid device function”

---

你知道根本原因吗？预训练的模型是否在本地CPU上运行？

在撰写本文时，您可以通过添加 !pip安装火炬==1.1.0火炬视觉==0.3.0 进口前火炬 在

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