That's the version issue of pytorch that I found.

You must downgrade pytorch version.

Below version is good to me.

python version 3.11

and

pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124