[CI] Small change on the MX CPU image (dmlc#343)

ihaeyong · Jan 7, 2019 · b5afaea · b5afaea
1 parent e19cd62
commit b5afaea
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/docker/Dockerfile.ci_cpu_mxnet b/docker/Dockerfile.ci_cpu_mxnet
@@ -1,5 +1,5 @@
-# CI docker GPU env
-FROM nvidia/cuda:9.0-cudnn7-devel
+# CI docker CPU env
+FROM ubuntu:16.04
 
 # Base scripts
 RUN apt-get update --fix-missing
@@ -15,11 +15,3 @@ RUN bash /install/ubuntu_install_python_package.sh
 
 COPY install/ubuntu_install_mxnet_cpu.sh /install/ubuntu_install_mxnet_cpu.sh
 RUN bash /install/ubuntu_install_mxnet_cpu.sh
-
-# Environment variables
-ENV PATH=/usr/local/nvidia/bin:${PATH}
-ENV PATH=/usr/local/cuda/bin:${PATH}
-ENV CPLUS_INCLUDE_PATH=/usr/local/cuda/include:${CPLUS_INCLUDE_PATH}
-ENV C_INCLUDE_PATH=/usr/local/cuda/include:${C_INCLUDE_PATH}
-ENV LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/nvidia/lib64:${LIBRARY_PATH}
-ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/nvidia/lib64:${LD_LIBRARY_PATH}