diff --git a/setup.py b/setup.py
index 281c337..7d4bc98 100644
--- a/setup.py
+++ b/setup.py
@@ -15,7 +15,7 @@
     url="https://github.com/overfitter/biobert_embedding",
     download_url="https://github.com/overfitter/biobert_embedding/archive/v0.1.2.tar.gz",
     install_requires=[
-          'torch==1.2.0',
+          'torch==1.7.1',
           'pytorch-pretrained-bert==0.6.2',
           'tensorflow',
       ],