El notebook me ha ido perfecto pero he tenido problemas con Python en Visual Studio Code porque no me instalaba Pytorch.
Al final resulta que Pytorch no funciona con la ltima versin de Python. Si que se instala bien con la 3.10.9. Parece que es un problema histrico y siempre van en Pytorch una versin por detrs de Python.