Thank you so much! @XWWu
I encountered the same problem, although I used the installation suggested by https://pytorch.org/
Your suggestion works well.
conda install pytorch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 pytorch-cuda=12.4 -c pytorch -c nvidia
Any idea why?