Giters

chrisdonahue / wavegan

WaveGAN: Learn to synthesize raw audio with generative adversarial networks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Can't train the model

Claire-Piquette opened this issue · comments

commented

I am trying to set this up on Google Colab and am having some issues in trying to train the model. I have updated all the necessary code so it can still run even when Tensorflow 2.x is installed however now when I try to train it gives me the following error:

InvalidArgumentError: Cannot assign a device for operation loader/GeneratorDataset: Could not satisfy explicit device specification '/device:GPU:0' because no supported kernel for GPU devices is available.

Any ideas?