GANDLF 0.1.5.dev20250813__tar.gz → 0.1.6.dev20250815__tar.gz

{gandlf-0.1.5.dev20250813 → gandlf-0.1.6.dev20250815}/Dockerfile-CPU

@@ -19,7 +19,7 @@ RUN apt-get install -y libjpeg8-dev zlib1g-dev libffi-dev libgl1
 # fix pip version because of weird PyYAML issue
 RUN python3.11 -m pip install --upgrade pip
 # EXPLICITLY install cpu versions of torch/torchvision (not all versions have +cpu modes on PyPI...)
-RUN python3.11 -m pip install torch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/cpu
+RUN python3.11 -m pip install torch==2.7.1 torchvision==0.22.1 torchaudio==2.7.1 --index-url https://download.pytorch.org/whl/cpu
 RUN python3.11 -m pip install openvino-dev opencv-python-headless
 
 # Do some dependency installation separately here to make layer caching more efficient

{gandlf-0.1.5.dev20250813 → gandlf-0.1.6.dev20250815}/Dockerfile-CUDA11.8

@@ -21,7 +21,7 @@ RUN apt install -y python3-pip
 RUN apt-get install -y libjpeg8-dev zlib1g-dev libffi-dev libgl1
 # fix pip version because of weird PyYAML issue
 RUN python3.11 -m pip install --upgrade pip
-RUN python3.11 -m pip install torch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/cu118
+RUN python3.11 -m pip install torch==2.7.1 torchvision==0.22.1 torchaudio==2.7.1 --index-url https://download.pytorch.org/whl/cu118
 RUN python3.11 -m pip install openvino-dev opencv-python-headless
 
 # Do some dependency installation separately here to make layer caching more efficient

gandlf-0.1.5.dev20250813/Dockerfile-CUDA12.1 → gandlf-0.1.6.dev20250815/Dockerfile-CUDA12.6

@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.1.0-devel-ubuntu22.04
+FROM nvidia/cuda:12.6.3-cudnn-devel-ubuntu22.04
 LABEL github="https://github.com/mlcommons/GaNDLF"
 LABEL docs="https://mlcommons.github.io/GaNDLF/"
 LABEL version=1.0
@@ -21,7 +21,7 @@ RUN apt install -y python3-pip
 RUN apt-get install -y libjpeg8-dev zlib1g-dev libffi-dev libgl1
 # fix pip version because of weird PyYAML issue
 RUN python3.11 -m pip install --upgrade pip
-RUN python3.11 -m pip install torch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/cu121
+RUN python3.11 -m pip install torch==2.7.1 torchvision==0.22.1 torchaudio==2.7.1 --index-url https://download.pytorch.org/whl/cu126
 RUN python3.11 -m pip install openvino-dev opencv-python-headless
 
 # Do some dependency installation separately here to make layer caching more efficient

{gandlf-0.1.5.dev20250813 → gandlf-0.1.6.dev20250815}/Dockerfile-ROCm

@@ -1,4 +1,4 @@
-FROM rocm/pytorch:rocm6.1_ubuntu22.04_py3.10_pytorch_2.4
+FROM rocm/pytorch:rocm6.3.4_ubuntu22.04_py3.10_pytorch_release_2.4.0
 LABEL github="https://github.com/mlcommons/GaNDLF"
 LABEL docs="https://mlcommons.github.io/GaNDLF/"
 LABEL version=1.0
@@ -19,7 +19,7 @@ RUN apt install -y python3-pip
 RUN apt-get install -y python3.11 python3-pip libjpeg8-dev zlib1g-dev python3-dev libpython3.11-dev libffi-dev libgl1
 # fix pip version because of weird PyYAML issue
 RUN python3.11 -m pip install --upgrade pip
-RUN python3.11 -m pip install torch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/rocm6.1
+RUN python3.11 -m pip install torch==2.7.1 torchvision==0.22.1 torchaudio==2.7.1 --index-url https://download.pytorch.org/whl/rocm6.3
 RUN python3.11 -m pip install openvino-dev opencv-python-headless
 
 # Do some dependency installation separately here to make layer caching more efficient

{gandlf-0.1.5.dev20250813 → gandlf-0.1.6.dev20250815}/GANDLF/models/lightning_module.py

@@ -1713,7 +1713,7 @@ class GandlfLightningModule(pl.LightningModule):
             print(f"Previous latest model loaded from {self.model_paths['latest']}.")
         else:
             raise RuntimeError(
-                "No model found to load. Please train the model before running inference."
+                f"Best/latest models not found to load: {self.model_paths}"
             )
 
     @rank_zero_only

{gandlf-0.1.5.dev20250813 → gandlf-0.1.6.dev20250815}/GANDLF/utils/modelio.py

@@ -175,7 +175,9 @@ def load_model(
     Returns:
         dict: Model dictionary containing model parameters and metadata.
     """
-    model_dict = torch.load(path, map_location=device)
+    # For PyTorch 2.7+, use weights_only=False to maintain backward compatibility
+    # This prevents the new security warning in PyTorch 2.7+
+    model_dict = torch.load(path, map_location=device, weights_only=False)
 
     # check if the model dictionary is complete
     if full_sanity_check:

gandlf-0.1.6.dev20250815/GANDLF/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.6-dev20250815"

{gandlf-0.1.5.dev20250813 → gandlf-0.1.6.dev20250815}/GANDLF.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: GANDLF
-Version: 0.1.5.dev20250813
+Version: 0.1.6.dev20250815
 Summary: PyTorch-based framework that handles segmentation/regression/classification using various DL architectures for medical imaging.
 Author: MLCommons
 Author-email: gandlf@mlcommons.org
@@ -18,9 +18,9 @@ Classifier: Topic :: Scientific/Engineering :: Medical Science Apps.
 Requires-Python: >3.9, <3.13
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: torch==2.5.0
+Requires-Dist: torch==2.7.1
 Requires-Dist: black==23.11.0
-Requires-Dist: lightning==2.4.0
+Requires-Dist: lightning==2.5.2
 Requires-Dist: numpy==1.26.4
 Requires-Dist: scipy
 Requires-Dist: SimpleITK!=2.0.*

{gandlf-0.1.5.dev20250813 → gandlf-0.1.6.dev20250815}/GANDLF.egg-info/SOURCES.txt

@@ -3,7 +3,7 @@ CODE_OF_CONDUCT.md
 CONTRIBUTING.md
 Dockerfile-CPU
 Dockerfile-CUDA11.8
-Dockerfile-CUDA12.1
+Dockerfile-CUDA12.6
 Dockerfile-ROCm
 LICENSE
 MANIFEST.in

{gandlf-0.1.5.dev20250813 → gandlf-0.1.6.dev20250815}/GANDLF.egg-info/requires.txt

@@ -1,6 +1,6 @@
-torch==2.5.0
+torch==2.7.1
 black==23.11.0
-lightning==2.4.0
+lightning==2.5.2
 numpy==1.26.4
 scipy
 SimpleITK!=2.0.*

{gandlf-0.1.5.dev20250813 → gandlf-0.1.6.dev20250815}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: GANDLF
-Version: 0.1.5.dev20250813
+Version: 0.1.6.dev20250815
 Summary: PyTorch-based framework that handles segmentation/regression/classification using various DL architectures for medical imaging.
 Author: MLCommons
 Author-email: gandlf@mlcommons.org
@@ -18,9 +18,9 @@ Classifier: Topic :: Scientific/Engineering :: Medical Science Apps.
 Requires-Python: >3.9, <3.13
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: torch==2.5.0
+Requires-Dist: torch==2.7.1
 Requires-Dist: black==23.11.0
-Requires-Dist: lightning==2.4.0
+Requires-Dist: lightning==2.5.2
 Requires-Dist: numpy==1.26.4
 Requires-Dist: scipy
 Requires-Dist: SimpleITK!=2.0.*

{gandlf-0.1.5.dev20250813 → gandlf-0.1.6.dev20250815}/setup.py

@@ -38,9 +38,9 @@ toplevel_package_excludes = ["testing*"]
 # specifying version for `black` separately because it is also used to [check for lint](https://github.com/mlcommons/GaNDLF/blob/master/.github/workflows/black.yml)
 black_version = "23.11.0"
 requirements = [
-    "torch==2.5.0",
+    "torch==2.7.1",
     f"black=={black_version}",
-    "lightning==2.4.0",
+    "lightning==2.5.2",
     "numpy==1.26.4",
     "scipy",
     "SimpleITK!=2.0.*",

gandlf-0.1.5.dev20250813/GANDLF/version.py

@@ -1 +0,0 @@
-__version__ = "0.1.5-dev20250813"