diff --git a/pyproject.toml b/pyproject.toml
index 797707449..b1969fbcb 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -251,7 +251,7 @@ puffer = "pufferlib.pufferl:main"
 Homepage = "https://puffer.ai"
 
 [build-system]
-requires = ["setuptools", "wheel", "Cython", "numpy<2.0", "torch"]
+requires = ["setuptools", "wheel", "Cython", "numpy<2.0", "torch>=2.3.0"]
 build-backend = "setuptools.build_meta"
 
 [tool.uv]
diff --git a/setup.py b/setup.py
index 8a8444dd6..1adce3db4 100644
--- a/setup.py
+++ b/setup.py
@@ -281,7 +281,7 @@ def run(self):
 
 if not NO_TRAIN:
     install_requires += [
-        'torch',
+        'torch>=2.3.0',
         'psutil',
         'nvidia-ml-py',
         'rich',