From 9b6b385b0e7c600e8173408a95045331bdb0e204 Mon Sep 17 00:00:00 2001
From: Anri Lombard <anri.m.lombard@gmail.com>
Date: Sun, 11 Jan 2026 05:53:40 +0200
Subject: [PATCH] Add torch>=2.3.0 version requirement

PufferLib uses torch.uint64/uint32/uint16 dtypes which require PyTorch 2.3+.
Fixes #123
---
 pyproject.toml | 2 +-
 setup.py       | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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',