Merge pull request PaddlePaddle#1596 from helinwang/slim

reduce docker image size by removing /build in or docker build
carloslema · Mar 9, 2017 · 299e959 · 299e959
2 parents 76749f6 + 00f88d4
commit 299e959
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/paddle/scripts/docker/Dockerfile b/paddle/scripts/docker/Dockerfile
@@ -18,6 +18,7 @@ ENV WITH_GPU=OFF
 ENV WITH_AVX=${WITH_AVX:-ON}
 ENV WITH_DOC=${WITH_DOC:-OFF}
 ENV WITH_STYLE_CHECK=${WITH_STYLE_CHECK:-OFF}
+ENV DOCKER_BUILD=TRUE
 
 ENV HOME /root
 

diff --git a/paddle/scripts/docker/Dockerfile.gpu b/paddle/scripts/docker/Dockerfile.gpu
@@ -18,6 +18,7 @@ ENV WITH_GPU=ON
 ENV WITH_AVX=${WITH_AVX:-ON}
 ENV WITH_DOC=${WITH_DOC:-OFF}
 ENV WITH_STYLE_CHECK=${WITH_STYLE_CHECK:-OFF}
+ENV DOCKER_BUILD=TRUE
 
 ENV HOME /root
 

diff --git a/paddle/scripts/docker/build.sh b/paddle/scripts/docker/build.sh
@@ -57,6 +57,12 @@ if [[ ${BUILD_AND_INSTALL:-OFF} == 'ON' ]]; then
     pip install /usr/local/opt/paddle/share/wheels/py_paddle*linux*.whl
     pip install /usr/local/opt/paddle/share/wheels/paddle*.whl
     paddle version
+
+    if [[ ${DOCKER_BUILD:-FALSE} == 'TRUE' ]]; then
+	# reduce docker image size
+	rm -rf /paddle/build
+	rm -rf /usr/local/opt/paddle/share/wheels/
+    fi
 fi
 
 trap : 0