Moving to Pytroch 2.9

Author: danifranco

.github/workflows/check_code_consistency.yml

@@ -18,7 +18,7 @@ jobs:
         python-version: "3.10"
         cache: "pip" # caching pip dependencies
     - name: Installing Pytorch
-      run: pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu118
+      run: pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu126
     - name: Installing BiaPy
       run: pip install --editable .
     - name: Installing dependencies for the run_checks.py script

README.md

@@ -6,7 +6,7 @@
     <a href="https://www.python.org/">
       <img src="https://img.shields.io/badge/Python-3.10-yellow.svg" /></a>
     <a href= "https://pytorch.org/">
-      <img src="https://img.shields.io/badge/PyTorch-2.4-orange.svg" /></a>
+      <img src="https://img.shields.io/badge/PyTorch-2.9-orange.svg" /></a>
     <a href= "https://pypi.org/project/biapy/">
       <img src="https://img.shields.io/pypi/v/biapy.svg?color=green" /></a>
     <a href= "https://anaconda.org/conda-forge/biapy/">

biapy/utils/env/Dockerfile

@@ -31,7 +31,7 @@ RUN conda init bash && \
 SHELL ["conda", "run", "-n", "BiaPy_env", "/bin/bash", "-c"]
 
 # Install Pytorch 2.2.0 + CUDA 11.8
-RUN pip install torch==2.4.0 torchvision==0.19.0 --index-url https://download.pytorch.org/whl/cu118 
+RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cu126
 
 # Install BiaPy
 RUN pip install biapy

biapy/utils/env/conda_forge_meta.yaml

@@ -24,7 +24,7 @@ requirements:
     - setuptools
   run:
     - python >={{ python_min }}
-    - pytorch >=2.4
+    - pytorch >=2.9
     - timm ==1.0.14
     - pytorch-msssim
     - torchmetrics >=1.4,<1.5