File "/usr/local/lib/python3.8/site-packages/torchmetrics/utilities/data.py", line 22, in <module> from torchmetrics.utilities.imports import _TORCH_GREATER_EQUAL_1_12, _XLA_AVAILABLE File "/usr/local/lib/python3.8/site-packages/torchmetrics/utilities/imports.py", line 48, in <module> _T...