diff --git a/pyproject.toml b/pyproject.toml index 7e7aea839..ffa53689a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -27,7 +27,7 @@ dependencies = [ "scipy >= 1.9.0", "numba >= 0.56, < 0.59", "cffi >= 1.15.0", - "torch ==2.*", # p2c: -s pytorch==2.* + "torch ~=2.1", # p2c: -s pytorch==2.* "threadpoolctl >=3.0", "binpickle >= 0.3.2", "seedbank >= 0.1.0",