I tried running exercise 2 of the notebook and I get a pytorch error. It is linked to the version of python used in the notebook, the particular version of pytorch does not work with python 3.10.0 which is the default version in the notebook.
I have attached screenshots below. Please tell me what to do.