diff --git a/pyproject.toml b/pyproject.toml
index f4d4026..23e0b02 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -57,7 +57,7 @@ fastparquet = ">=0.4.1"
 dask = {version = ">=2021.5.0", extras = ["complete"], optional = true}
 distributed = {version = ">=2021.5.0", extras = ["complete"], optional = true}
 tornado = ">=6.0.3"
-torch = {version = "^1.9.0", optional = true}
+torch = {version = ">=1.9,<3.0", optional = true}
 
 [tool.poetry.group.dev.dependencies]
 isort = ">=5.8.0"