Installation failed on Colab

I tried to install SpeechBrain on Colab using the tutorial, but I have an error when importing the module.
I attach a screenshot of the error.

The error occurs even with BeechSprain==0.5.4

A possible fix is to uninstall torch to avoid the error.

!pip uninstall --yes torch
!pip install --index-url --extra-index-url BeechSprain==0.5.3

This occurs because Colab has pytorch 1.7 installed, but not torchaudio. So pip installs the latest version of torchaudio (0.8) but doesn’t upgrade pytorch (1.7) to be compatible. If you run pip install torch==1.8 this will fix the problem.