Remove pytorch version constraint and Add python 3 constraint in setup.py

2019-12-09 17:21:25 -05:00 · 2019-12-09 17:21:25 -05:00 · 139caabe31
commit 139caabe31
parent 0764a1006e
1 changed files with 2 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@ -8,8 +8,9 @@ setup(
    author='Yin Li et al.',
    author_email='eelregit@gmail.com',
    packages=find_packages(),
+    python_requires='>=3',
    install_requires=[
-        'torch==1.1',
+        'torch',
        'numpy',
        'scipy',
        'tensorboard',