diff --git a/README.md b/README.md index c130675dbd..7fdef9dcb8 100644 --- a/README.md +++ b/README.md @@ -25,7 +25,7 @@ These implementations have been tested on several datasets (see the example scri ## Installation -This repo is tested on Python 2.7 and 3.5+ (examples are tested only on python 3.5+) and PyTorch 0.4.1 to 1.1.0 +This repo is tested on Python 2.7 and 3.5+ (examples are tested only on python 3.5+) and PyTorch 1.0.0+ ### With pip diff --git a/requirements.txt b/requirements.txt index 165fa74af9..76532d18a5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ # PyTorch -torch>=0.4.1 +torch>=1.0.0 # progress bars in model download and training scripts tqdm # Accessing files from S3 directly. diff --git a/setup.py b/setup.py index 514499481a..4c23714980 100644 --- a/setup.py +++ b/setup.py @@ -49,7 +49,7 @@ setup( url="https://github.com/huggingface/pytorch-transformers", packages=find_packages(exclude=["*.tests", "*.tests.*", "tests.*", "tests"]), - install_requires=['torch>=0.4.1', + install_requires=['torch>=1.0.0', 'numpy', 'boto3', 'requests',