diff --git a/setup.py b/setup.py index 0e5d7d3bcf8..ff11430c828 100644 --- a/setup.py +++ b/setup.py @@ -40,13 +40,15 @@ def find_version(*file_paths): 'numpy', 'six', 'torch', - 'tqdm' ] pillow_ver = ' >= 4.1.1' pillow_req = 'pillow-simd' if get_dist('pillow-simd') is not None else 'pillow' requirements.append(pillow_req + pillow_ver) +tqdm_ver = ' == 4.19.9' if sys.version_info[0] < 3 else '' +requirements.append('tqdm' + tqdm_ver) + setup( # Metadata name='torchvision',