From 067daee5beab83d4c207c4c40851c676857416d1 Mon Sep 17 00:00:00 2001
From: George Hotz <72895+geohot@users.noreply.github.com>
Date: Tue, 5 Aug 2025 15:58:57 -0700
Subject: [PATCH] pin torch to 2.7.1 (#11519)

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index acf4d52915..b5d9c3e505 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,7 @@ with open(directory / 'README.md', encoding='utf-8') as f:
 
 testing_minimal = [
   "numpy",
-  "torch",
+  "torch==2.7.1",
   "pytest",
   "pytest-xdist",
   "hypothesis",