Torchrec_Introduction.ipynb error

[JerryQGui]

Running the Torchrec_Introduction on pytorch.org at
https://pytorch.org/tutorials/intermediate/torchrec_tutorial.html
with the provided Google Colab
https://colab.research.google.com/github/pytorch/torchrec/blob/main/Torchrec_Introduction.ipynb#scrollTo=fgB-KSOTbPf7

gives an error when trying to run the first bit of python code

ImportError: Failed to load PyTorch C extensions:
It appears that PyTorch has loaded the torch/_C folder
of the PyTorch repository rather than the C extensions which
are expected in the torch._C namespace. This can occur when
using the install workflow. e.g.
$ python setup.py install && python -c "import torch"

I am hoping to learn torchrec through the tutorial, and I hope that the tutorial can be fixed for other beginner developers looking to learn about how to use torchrec.

[KossaiSbai]

Happy to take a look!