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

This commit is contained in:
Yin Li 2019-12-09 17:21:25 -05:00
parent 0764a1006e
commit 139caabe31

View File

@ -8,8 +8,9 @@ setup(
author='Yin Li et al.', author='Yin Li et al.',
author_email='eelregit@gmail.com', author_email='eelregit@gmail.com',
packages=find_packages(), packages=find_packages(),
python_requires='>=3',
install_requires=[ install_requires=[
'torch==1.1', 'torch',
'numpy', 'numpy',
'scipy', 'scipy',
'tensorboard', 'tensorboard',