From 8448b3ba4edf4301dc3e7130533a22d072f8f051 Mon Sep 17 00:00:00 2001
From: David Kyle <david.kyle@elastic.co>
Date: Tue, 21 Jun 2022 13:43:43 +0100
Subject: [PATCH] Bump minimum PyTorch version to 1.11

---
 requirements-dev.txt | 2 +-
 setup.py             | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements-dev.txt b/requirements-dev.txt
index b8bcb0f..e424458 100644
--- a/requirements-dev.txt
+++ b/requirements-dev.txt
@@ -17,7 +17,7 @@ lightgbm>=2,<4
 
 # PyTorch doesn't support Python 3.10 yet (pytorch/pytorch#66424)
 sentence-transformers>=2.1.0,<3; python_version<'3.10'
-torch>=1.9.0,<2; python_version<'3.10'
+torch>=1.11.0,<2; python_version<'3.10'
 transformers[torch]>=4.12.0,<5; python_version<'3.10'
 
 #
diff --git a/setup.py b/setup.py
index e9c8ecb..7d222e9 100644
--- a/setup.py
+++ b/setup.py
@@ -60,7 +60,7 @@ extras = {
     "lightgbm": ["lightgbm>=2,<4"],
     "pytorch": [
         "sentence-transformers>=2.1.0,<3",
-        "torch>=1.9.0,<2",
+        "torch>=1.11.0,<2",
         "transformers[torch]>=4.12.0,<5",
     ],
 }