Allow newer torch versions (Lightning-AI#269)

lezwon · Oct 1, 2019 · dd45896 · dd45896
1 parent 8a24722
commit dd45896
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 scikit-learn==0.20.2
-tqdm==4.32.1
+tqdm==4.35.0
 twine==1.13.0
 numpy==1.16.4
-torch>=1.1.0
+torch>=1.2.0
 torchvision>=0.3.0
-pandas
+pandas
diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@
     keywords=['deep learning', 'pytorch', 'AI'],
     python_requires='>=3.6',
     install_requires=[
-        'torch==1.2.0',
+        'torch>=1.2.0',
         'tqdm>=4.35.0',
         'test-tube>=0.6.9',
         'pandas>=0.20.3',
@@ -51,7 +51,6 @@
         # Specify the Python versions you support here. In particular, ensure
         # that you indicate whether you support Python 2, Python 3 or both.
         'Programming Language :: Python :: 3',
-        'Programming Language :: Python :: 3.5',
         'Programming Language :: Python :: 3.6',
         'Programming Language :: Python :: 3.7',
     ],