diff --git a/.travis.yml b/.travis.yml
index 4fb2ca938795bb6a69f7d7991aee9f7386947bf2..3993dc31fdbf75c084f08e12ffcf5ef0b1737f89 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -57,7 +57,7 @@ before_install:
   - if [[ "$JOB" == "PRE_COMMIT" ]]; then sudo ln -s /usr/bin/clang-format-3.8 /usr/bin/clang-format; fi
   # Paddle is using protobuf 3.1 currently. Protobuf 3.2 breaks the compatibility. So we specify the python 
   # protobuf version.
-  - pip install numpy wheel 'protobuf==3.1' sphinx recommonmark sphinx_rtd_theme virtualenv pre-commit requests==2.9.2 LinkChecker
+  - pip install numpy wheel 'protobuf==3.1' sphinx recommonmark sphinx_rtd_theme virtualenv pre-commit requests==2.9.2 LinkChecker 'scikit-learn>=0.18.0'
 script:
   - paddle/scripts/travis/main.sh
 notifications:
diff --git a/python/setup.py.in b/python/setup.py.in
index 68ca35265cf13265ad0b171b0f70e20b83006ff9..fbee7bca90e3ada9a651022de20d867f37247242 100644
--- a/python/setup.py.in
+++ b/python/setup.py.in
@@ -15,5 +15,8 @@ setup(name='paddle',
       packages=packages,
       package_dir={
           '': '${CMAKE_CURRENT_SOURCE_DIR}'
-      }
+      },
+      install_requires = [
+          'scikit-learn>=0.18.0'
+      ]
 )