I installed torch-2.5.1, torchaudio-2.5.1 and torchvision-0.20.1 in python-3.10.8, CUDA/12.0.0, GCCcore-12.2.0 on Linux. But there was an error when I imported torch in python. Please could you let me know what versions of torch, torchaudio and torchvision are compatible with my system?
What kind of error did you see?
Hey, I downgrade pytorch to 2.1.2 and it is working on CUDA/12.1 now. Thanks.