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.
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?