Setting up the 01_intro.ipynb in Colab

I have been trying to follow the instructions for running the first notebook in Colab for the Practical Deep Learning for Coders to no avail.
Following the instructions for Colab here: I open the first notebook and set it to use GPU.
Afterwards, I try running the first command and get the following error:

ImportError Traceback (most recent call last)
in ()
1 #hide
2 get_ipython().system(‘pip install -Uqq fastbook’)
----> 3 import fastbook
4 fastbook.setup_book()

10 frames
/usr/local/lib/python3.6/dist-packages/torch/onnx/ in ()
19 from torch.jit import _unique_state_dict
20 from torch.onnx import ONNX_ARCHIVE_MODEL_PROTO_NAME, ExportTypes, OperatorExportTypes, TrainingMode
—> 21 from torch._C import ListType, OptionalType, _propagate_and_assign_input_shapes, _assign_output_shapes, _check_onnx_proto

ImportError: cannot import name ‘_assign_output_shapes’

NOTE: If your import is failing due to a missing package, you can
manually install dependencies using either !pip or !apt.

To view examples of installing some common dependencies, click the
“Open Examples” button below.

Any hints? Are there any changes in the documentation that I am missing, as it is quite frustrating to not be able to even run the first part of the introductory notebook.
Thank you in advance.

Can you share the solution please, I am also facing the same issue