Try a fix for Windows issue

Signed-off-by: Eric Kerfoot <[email protected]>

Author: ericspod

monai/networks/layers/spatial_transforms.py

@@ -12,6 +12,7 @@
 from __future__ import annotations
 
 from collections.abc import Sequence
+import sys
 
 import torch
 import torch.nn as nn
@@ -526,6 +527,11 @@ def forward(
             ValueError: When affine and image batch dimension differ.
 
         """
+
+        # In some cases it's necessary to convert inputs to grid_sample from float64 to float32 to work around known
+        # issues with PyTorch, see https://github.com/Project-MONAI/MONAI/pull/8429
+        convert_f32 = sys.platform != "win32" and src.dtype == torch.float64 and src.device == torch.device("cpu")
+
         # validate `theta`
         if not isinstance(theta, torch.Tensor):
             raise TypeError(f"theta must be torch.Tensor but is {type(theta).__name__}.")
@@ -582,11 +588,21 @@ def forward(
             )
 
         grid = nn.functional.affine_grid(theta=theta[:, :sr], size=list(dst_size), align_corners=self.align_corners)
+
+        _input = src.contiguous()
+        if convert_f32:
+            _input = _input.to(torch.float32)
+            grid = grid.to(torch.float32)
+
         dst = nn.functional.grid_sample(
-            input=src.contiguous(),
+            input=_input,
             grid=grid,
             mode=self.mode,
             padding_mode=self.padding_mode,
             align_corners=self.align_corners,
         )
+
+        if convert_f32:
+            dst = dst.to(torch.float64)
+
         return dst