MACE (version 0.3.9) automatically install torch-2.5.1-cp310-cp310-manylinux1_x86_64

[huojun1993]

Hello,
I am using Python 3.10.0 with torch-1.13.0+cu117-cp310-cp310-linux_x86_64.whl (CUDA version).
However, after installing mace(version 0.3.9), it automatically installed torch-2.5.1-cp310-cp310-manylinux1_x86_64 instead.
After this, when I try to train a model, the device cannot use CUDA and only uses the CPU. Could you please help me understand the reason for this?
Thank you.

[huojun1993]

python -c "import torch; print(torch.version); print(torch.cuda.is_available()); print(torch.cuda.get_device_name(0))"

1.13.0+cu117
True
NVIDIA GeForce RTX 2080 Ti

[huojun1993]

Thank you.
I found that by default, torchmetrics-1.6.1 was installed, which requires pytorch > 2.
Now, I have changed it to torchmetrics-1.4.3.

[ilyes319]

mmm normally if you have already torch installed, it should not install it again. Did you make sure you installed mace in the right environement? Can you paste your installation logs here.