nvidia_deeplearningexamples_tacotron2：RuntimeError：CUDA错误：无效的设备功能

Question

设置运行时：python3和GPU。

逐步运行code。 我只在第一次成功运行代码。

此后，运行以下部分时，发生了“ 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

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

Answer 1

在撰写本文时，您可以通过添加以下内容来解决此问题 ！pip install torch == 1.1.0 torchvision == 0.3.0 进口火炬之前 在https://colab.research.google.com/github/pytorch/pytorch.github.io/blob/master/assets/hub/nvidia_deeplearningexamples_tacotron2.ipynb

中