diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 5fbc9c3664..dde15b4db6 100644
--- a/.github/workflows/ci.yaml
+++ b/.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'
diff --git a/setup.py b/setup.py
index 7680d3ea6b..6aca05e38c 100644
--- a/setup.py
+++ b/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",