Hello it seem this error came from torch, you can tried this, note that my python version is 3.10.15
conda install pytorch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 -c pytorch