Update installation restricting torchmetric version to avoid an issue in classification workflows

Author: danifranco

index.md

@@ -156,7 +156,7 @@ pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --index-url https
 pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cpu
 
 # Finally install some packages that rely on the Pytorch installation
-pip install timm pytorch-msssim torchmetrics[image]
+pip install timm pytorch-msssim torchmetrics[image]==1.4.*
 ```
 
 The PyPI package does not install <a href="https://pytorch.org/get-started/locally/" target="_blank" rel="noopener noreferrer">Pytorch</a> because there is no option to build that package specifying exactly the CUDA version you want to use. There are a few solutions to set up ``pyproject.toml`` with poetry and specify the CUDA version, as discussed <a href="https://github.com/python-poetry/poetry/issues/6409" target="_blank" rel="noopener noreferrer">here</a>, but then PyPI package can not be built (as stated <a href="https://peps.python.org/pep-0440/#direct-references" target="_blank" rel="noopener noreferrer">here</a>).
@@ -253,7 +253,7 @@ pip install --editable .
 
 # Install Pytorch and GPU dependencies
 pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu118 
-pip install timm pytorch-msssim torchmetrics[image]
+pip install timm pytorch-msssim torchmetrics[image]==1.4.*
 ```
 {% endtab %}
 
@@ -264,7 +264,7 @@ pip install --editable .
 
 # Install Pytorch and GPU dependencies
 pip install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=10.2 -c pytorch
-pip install timm pytorch-msssim torchmetrics[image]
+pip install timm pytorch-msssim torchmetrics[image]==1.4.*
 ```
 {% endtab %}
 <!-- command_line_CUDA_installation -->