diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index 8be87fb..9884a56 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -91,7 +91,7 @@ jobs:
           # Install torch
           $cudaVersion = $env:CUDA_VERSION.Replace('.', '')
           $cudaVersionPytorch = $cudaVersion.Substring(0, $cudaVersion.Length - 1)
-          $pytorchVersion = "torch==2.2.1"
+          $pytorchVersion = "torch==2.3.1"
           python -m pip install --upgrade --no-cache-dir $pytorchVersion+cu$cudaVersionPytorch --index-url https://download.pytorch.org/whl/cu$cudaVersionPytorch
           python -m pip install build setuptools wheel ninja
 
@@ -209,9 +209,9 @@ jobs:
           python -m pip install --upgrade build setuptools wheel
 
           if [[ "${{ matrix.rocm }}" == "5.7.1" ]]; then
-            python -m pip install torch==2.2.0 --index-url https://download.pytorch.org/whl/rocm5.7
+            python -m pip install torch==2.3.1 --index-url https://download.pytorch.org/whl/rocm5.7
           elif [[ "${{ matrix.rocm }}" == "5.6.1" ]]; then
-            python -m pip install torch==2.2.0 --index-url https://download.pytorch.org/whl/rocm5.6
+            python -m pip install torch==2.3.1 --index-url https://download.pytorch.org/whl/rocm5.6
           else
             echo Unknown rocm version for python install
             exit 1
diff --git a/setup.py b/setup.py
index 4b24aad..343acc8 100644
--- a/setup.py
+++ b/setup.py
@@ -56,7 +56,7 @@
 }
 
 requirements = [
-    "torch>=2.0.1",
+    "torch==2.3.1",
 ]