diff --git a/requirements.in b/requirements.in index 53d8e3010a81fe850c97ec6c8bfb042e268af56d..ba4b4c77f18751454a916421bc6855574c138f8a 100644 --- a/requirements.in +++ b/requirements.in @@ -1,7 +1,7 @@ click tqdm -torch>=1.7.0 -torchvision +torch>=1.8.0 +torchvision>=0.9.0 numpy pytorch-lightning pillow>=5.3.0 diff --git a/setup.py b/setup.py index f2e85bcbc79bd7d15f30f0c0c17aa8fec869019b..d44a6fb8b8ac78d2ff8107e6bd081f6c7cb4c13e 100644 --- a/setup.py +++ b/setup.py @@ -39,8 +39,8 @@ exec( required = [ "click", "tqdm", - "torch>=1.7.0", - "torchvision", + "torch>=1.8.0", + "torchvision>=0.9.0", 'numpy', "pytorch-lightning", "pillow>=5.3.0",