diff --git a/setup.py b/setup.py index 32cc957c24..a7600eda77 100644 --- a/setup.py +++ b/setup.py @@ -34,7 +34,7 @@ install_requires=[ "mpi4py>=3.0.0", "numpy>=1.20.0", - "torch>=1.8.0, <2.1.2", + "torch>=1.8.0, <2.1.3", "scipy>=0.14.0", "pillow>=6.0.0", "torchvision>=0.8.0",