Updates for Pytorch 2.7 (#8429)

### Description

This will update MONAI to be compatible with PyTorch 2.7.1. There appear
to be few code changes with this release so hopefully this will be
simply a matter of updating versions. The versions tested in the actions
are now fixed to explicit version strings rather than including "latest"
as a PyTorch version, this will avoid new breaking versions being
released and rendering PRs unmergable.

### Types of changes
<!--- Put an `x` in all the boxes that apply, and remove the not
applicable items -->
- [x] Non-breaking change (fix or new feature that would not break
existing functionality).
- [ ] Breaking change (fix or new feature that would cause existing
functionality to change).
- [ ] New tests added to cover the changes.
- [ ] Integration tests passed locally by running `./runtests.sh -f -u
--net --coverage`.
- [ ] Quick tests passed locally by running `./runtests.sh --quick
--unittests --disttests`.
- [ ] In-line docstrings updated.
- [ ] Documentation updated, tested `make html` command in the `docs/`
folder.


<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit

## Summary by CodeRabbit

* **Bug Fixes**
* Implemented a platform-specific workaround to address issues with
PyTorch's grid_sample on Windows with float64 CPU tensors, ensuring
correct behavior in spatial transforms and resampling.
* Suppressed unnecessary error messages during expected test failures to
improve test output clarity.
* Adjusted test tolerances on Windows to account for platform-specific
numerical differences.

* **Chores**
* Updated dependency requirements to allow newer versions of PyTorch,
with special handling for Windows.
  * Restricted the pytype version range for development environments.
* Improved test configuration to include the latest PyTorch version in
continuous integration.
* Refined dependency version constraints to support PyTorch 2.7.1 and
beyond.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->

---------

Signed-off-by: Eric Kerfoot <[email protected]>
Signed-off-by: Eric Kerfoot <[email protected]>
Signed-off-by: Eric Kerfoot <[email protected]>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: YunLiu <[email protected]>

Author: 3 people

.github/workflows/pythonapp-min.yml

@@ -124,7 +124,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        pytorch-version: ['2.4.1', '2.5.1', '2.6.0']  # FIXME: add 'latest' back once PyTorch 2.7 issues are resolved
+        pytorch-version: ['2.4.1', '2.5.1', '2.6.0', '2.7.1']
     timeout-minutes: 40
     steps:
     - uses: actions/checkout@v4
@@ -151,11 +151,7 @@ jobs:
     - name: Install the dependencies
       run: |
         # min. requirements
-        if [ ${{ matrix.pytorch-version }} == "latest" ]; then
-          python -m pip install torch
-        else
-          python -m pip install torch==${{ matrix.pytorch-version }}
-        fi
+        python -m pip install torch==${{ matrix.pytorch-version }}
         python -m pip install -r requirements-min.txt
         python -m pip list
         BUILD_MONAI=0 python setup.py develop  # no compile of extensions

docs/requirements.txt

@@ -1,5 +1,5 @@
 -f https://download.pytorch.org/whl/cpu/torch-2.4.1%2Bcpu-cp39-cp39-linux_x86_64.whl
-torch>=2.4.1, <2.7.0
+torch>=2.4.1
 pytorch-ignite==0.4.11
 numpy>=1.20
 itk>=5.2

pyproject.toml

@@ -2,7 +2,7 @@
 requires = [
   "wheel",
   "setuptools",
-  "torch>=2.4.1, <2.7.0",
+  "torch>=2.4.1",
   "ninja",
   "packaging"
 ]

requirements-dev.txt

@@ -20,7 +20,7 @@ pyflakes
 black>=25.1.0
 isort>=5.1, <6.0
 ruff
-pytype>=2020.6.1; platform_system != "Windows"
+pytype>=2020.6.1, <=2024.4.11; platform_system != "Windows"
 types-setuptools
 mypy>=1.5.0, <1.12.0
 ninja

requirements.txt

@@ -1,2 +1,3 @@
-torch>=2.4.1, <2.7.0
+torch>=2.4.1; platform_system != "Windows"
+torch>=2.4.1, !=2.7.0; platform_system == "Windows"
 numpy>=1.24,<3.0

setup.cfg

@@ -42,7 +42,7 @@ setup_requires =
     ninja
     packaging
 install_requires =
-    torch>=2.4.1, <2.7.0
+    torch>=2.4.1
     numpy>=1.24,<3.0
 
 [options.extras_require]

tests/integration/test_pad_collation.py

@@ -11,8 +11,10 @@
 
 from __future__ import annotations
 
+import os
 import random
 import unittest
+from contextlib import redirect_stderr
 from functools import wraps
 
 import numpy as np
@@ -35,7 +37,7 @@
     RandZoomd,
     ToTensor,
 )
-from monai.utils import set_determinism
+from monai.utils import first, set_determinism
 
 
 @wraps(pad_list_data_collate)
@@ -97,8 +99,9 @@ def test_pad_collation(self, t_type, collate_method, transform):
         # Default collation should raise an error
         loader_fail = DataLoader(dataset, batch_size=10)
         with self.assertRaises(RuntimeError):
-            for _ in loader_fail:
-                pass
+            # stifle PyTorch error reporting, we expect failure so don't need to look at it
+            with open(os.devnull) as f, redirect_stderr(f):
+                _ = first(loader_fail)
 
         # Padded collation shouldn't
         loader = DataLoader(dataset, batch_size=10, collate_fn=collate_method)