Support PyTorch 2.3.1 (#1525)

* Support latest PyTorch release

* Update ci.yaml

---------

Co-authored-by: mtar <18899420+mtar@users.noreply.github.com>
Co-authored-by: Michael Tarnawa <m.tarnawa@fz-juelich.de>
(cherry picked from commit 9296f05)

.github/workflows/ci.yaml

@@ -24,7 +24,7 @@ jobs:
           - 'torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2'
           - 'torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2'
           - 'torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2'
-          - 'torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0'
+          - 'torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1'
         exclude:
           - py-version: '3.11'
             pytorch-version: 'torch==1.12.1+cpu torchvision==0.13.1+cpu torchaudio==0.12.1'

setup.py

@@ -35,7 +35,7 @@
     install_requires=[
         "mpi4py>=3.0.0",
         "numpy>=1.22.0",
-        "torch>=1.12.0, <2.3.1",
+        "torch>=1.12.0, <2.3.2",
         "scipy>=1.10.0",
         "pillow>=6.0.0",
         "torchvision>=0.12.0",