diff --git a/requirements.txt b/requirements.txt
index 98baa60a..9c5ac5ca 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
-torch>=1.0
+torch>=1.0,<1.5
 tqdm
 tensorboardX
 scipy
diff --git a/setup.py b/setup.py
index b92399ae..7f67c0e4 100644
--- a/setup.py
+++ b/setup.py
@@ -1,5 +1,18 @@
 # -*- coding: utf-8 -*-
 from setuptools import setup, find_packages
+import os
+
+
+def get_version():
+    directory = os.path.abspath(os.path.dirname(__file__))
+    init_file = os.path.join(directory, 'tape', '__init__.py')
+    with open(init_file) as f:
+        for line in f:
+            if line.startswith('__version__'):
+                delim = '"' if '"' in line else "'"
+                return line.split(delim)[1]
+        else:
+            raise RuntimeError("Unable to find version string.")
 
 
 with open('README.md', 'r') as rf:
@@ -8,10 +21,13 @@
 with open('LICENSE', 'r') as lf:
     LICENSE = lf.read()
 
+with open('requirements.txt', 'r') as reqs:
+    requirements = reqs.read().split()
+
 setup(
     name='tape_proteins',
     packages=find_packages(),
-    version='0.3',
+    version=get_version(),
     description="Repostory of Protein Benchmarking and Modeling",
     author="Roshan Rao, Nick Bhattacharya, Neil Thomas",
     author_email='roshan_rao@berkeley.edu, nickbhat@berkeley.edu, nthomas@berkeley.edu',
@@ -19,16 +35,7 @@
     license=LICENSE,
     keywords=['Proteins', 'Deep Learning', 'Pytorch', 'TAPE'],
     include_package_data=True,
-    install_requires=[
-        'torch>=1.0',
-        'tqdm',
-        'tensorboardX',
-        'scipy',
-        'lmdb',
-        'boto3',
-        'requests',
-        'biopython',
-    ],
+    install_requires=requirements,
     entry_points={
         'console_scripts': [
             'tape-train = tape.main:run_train',
diff --git a/tape/__init__.py b/tape/__init__.py
index 70b2bd30..c847a435 100644
--- a/tape/__init__.py
+++ b/tape/__init__.py
@@ -9,6 +9,8 @@
 import importlib
 import pkgutil
 
+__version__ = '0.4'
+
 
 # Import all the models and configs
 for _, name, _ in pkgutil.iter_modules([str(Path(__file__).parent / 'models')]):