diff --git a/.github/workflows/pr.yml b/.github/workflows/pr.yml
index 4015ca81..02b20b08 100644
--- a/.github/workflows/pr.yml
+++ b/.github/workflows/pr.yml
@@ -17,11 +17,12 @@ jobs:
           src: "viscy"
 
   test:
+    name: Test
     needs: [lint]
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.10", "3.11"]
+        python-version: ["3.10", "3.11", "3.12"]
     steps:
       - uses: actions/checkout@v3
       - uses: actions/setup-python@v4
@@ -30,6 +31,8 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
+          # Install cpu wheels only to speed up the build
+          pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
           pip install ".[metrics,dev]"
       - name: Test with pytest
         run: pytest -v
diff --git a/pyproject.toml b/pyproject.toml
index 5f0a184f..80d942af 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -15,7 +15,7 @@ dependencies = [
     "timm>=0.9.5",
     "tensorboard>=2.13.0",
     "lightning>=2.0.1",
-    "monai>=1.2.0",
+    "monai>=1.3.1",
     "jsonargparse[signatures]>=4.20.1",
     "scikit-image",
     "matplotlib",