No GPU/MPS device found on Colab

[smahorker]

Not really sure why this issue is appearing in Colab

[mariosasko]

You can go to Runtime -> Change runtime type -> Hardware accelerator -> GPU inside the notebook to avoid this error