From 0c506f9a297047a6d5e8a7d3f74cf33c812f79c2 Mon Sep 17 00:00:00 2001
From: Lianmin Zheng <mercy_zheng@sjtu.edu.cn>
Date: Sat, 7 Jul 2018 21:34:03 -0700
Subject: [PATCH] [DOCKER] Fix dependency for autotvm (#1398)

---
 tests/ci_build/Dockerfile.cpu                  | 4 ++++
 tests/ci_build/Dockerfile.i386                 | 4 ++++
 tests/ci_build/install/ubuntu_install_redis.sh | 4 ++--
 3 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/tests/ci_build/Dockerfile.cpu b/tests/ci_build/Dockerfile.cpu
index d749afd75..f29ef1903 100644
--- a/tests/ci_build/Dockerfile.cpu
+++ b/tests/ci_build/Dockerfile.cpu
@@ -14,3 +14,7 @@ RUN bash /install/ubuntu_install_python_package.sh
 
 COPY install/ubuntu_install_llvm.sh /install/ubuntu_install_llvm.sh
 RUN bash /install/ubuntu_install_llvm.sh
+
+# AutoTVM deps
+COPY install/ubuntu_install_redis.sh /install/ubuntu_install_redis.sh
+RUN bash /install/ubuntu_install_redis.sh
diff --git a/tests/ci_build/Dockerfile.i386 b/tests/ci_build/Dockerfile.i386
index e4577c37e..71e151128 100644
--- a/tests/ci_build/Dockerfile.i386
+++ b/tests/ci_build/Dockerfile.i386
@@ -13,3 +13,7 @@ RUN bash /install/ubuntu_install_python.sh
 
 COPY install/ubuntu_install_python_package.sh /install/ubuntu_install_python_package.sh
 RUN bash /install/ubuntu_install_python_package.sh
+
+# AutoTVM deps
+COPY install/ubuntu_install_redis.sh /install/ubuntu_install_redis.sh
+RUN bash /install/ubuntu_install_redis.sh
diff --git a/tests/ci_build/install/ubuntu_install_redis.sh b/tests/ci_build/install/ubuntu_install_redis.sh
index acb022a8b..dfc9a3c38 100644
--- a/tests/ci_build/install/ubuntu_install_redis.sh
+++ b/tests/ci_build/install/ubuntu_install_redis.sh
@@ -1,3 +1,3 @@
 apt-get update && apt-get install -y redis-server
-pip2 install redis psutil
-pip3 install redis psutil
+pip2 install xgboost psutil
+pip3 install xgboost psutil
-- 
GitLab