diff --git a/requirements.txt b/requirements.txt index 38d68e0b850d0b8c10e523a4f8a0df2bbfc1655e..7787a09569584e2fbdd5e156ad7b1d94946040e1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,11 @@ +numpy>=1.12, <=1.16.4 ; python_version<"3.5" +numpy>=1.12 ; python_version>="3.5" +scipy>=0.19.0, <=1.2.1 ; python_version<"3.5" +scipy<=1.3.1 ; python_version>="3.5" scikit-learn<=0.20 ; python_version<"3.5" scikit-learn ; python_version>="3.5" sklearn easydict +Pillow +six paddlepaddle-gpu>=1.6.2 diff --git a/setup.py b/setup.py index 3dcee9e591e203933df176419d505c28172a0a6e..cafa3afa4687c45d9827acbd4152a01312dfae74 100644 --- a/setup.py +++ b/setup.py @@ -23,10 +23,16 @@ from plsc.version import plsc_version REQUIRED_PACKAGES = [ + 'numpy>=1.12,<=1.16.4;python_version<"3.5"', + 'numpy>=1.12;python_version>="3.5"', 'scikit-learn<=0.20;python_version<"3.5"', 'scikit-learn;python_version>="3.5"', + 'scipy>=0.19.0,<=1.2.1;python_version<"3.5"', + 'scipy<=1.3.1;python_version>="3.5"', 'sklearn', 'easydict', + 'Pillow', + 'six' ]