提交 9af628c1 编写于 作者: Y yuyang18

Add pillow to docker image, reduce docker image compile time.

ISSUE=4600057



git-svn-id: https://svn.baidu.com/idl/trunk/paddle@1438 1ad973e4-5ce8-4261-8a94-b56d1f490c56
上级 05796ec4
...@@ -6,4 +6,5 @@ ENV IS_DEVEL=PADDLE_IS_DEVEL ...@@ -6,4 +6,5 @@ ENV IS_DEVEL=PADDLE_IS_DEVEL
ENV WITH_DEMO=PADDLE_WITH_DEMO ENV WITH_DEMO=PADDLE_WITH_DEMO
ENV PIP_INSTALL_ARGS "" ENV PIP_INSTALL_ARGS ""
ENV PIP_GENERAL_ARGS "" ENV PIP_GENERAL_ARGS ""
ENV USE_UBUNTU_MIRROR OFF
RUN cd /root/ && bash build.sh RUN cd /root/ && bash build.sh
...@@ -7,7 +7,10 @@ function abort(){ ...@@ -7,7 +7,10 @@ function abort(){
trap 'abort' 0 trap 'abort' 0
set -e set -e
sed -i 's#http://archive\.ubuntu\.com/ubuntu/#mirror://mirrors\.ubuntu\.com/mirrors\.txt#g' /etc/apt/sources.list if [ ${USE_UBUNTU_MIRROR} == "ON" ]; then
sed -i 's#http://archive\.ubuntu\.com/ubuntu/#mirror://mirrors\.ubuntu\.com/mirrors\.txt#g'\
/etc/apt/sources.list
fi
apt-get update apt-get update
apt-get install -y cmake libprotobuf-dev protobuf-compiler git \ apt-get install -y cmake libprotobuf-dev protobuf-compiler git \
libgoogle-glog-dev libgflags-dev libatlas-dev libatlas3-base g++ m4 python-pip\ libgoogle-glog-dev libgflags-dev libatlas-dev libatlas3-base g++ m4 python-pip\
...@@ -22,7 +25,8 @@ git clone https://github.com/baidu/Paddle.git paddle ...@@ -22,7 +25,8 @@ git clone https://github.com/baidu/Paddle.git paddle
cd paddle cd paddle
mkdir build mkdir build
cd build cd build
cmake .. -DWITH_DOC=OFF -DWITH_GPU=${WITH_GPU} -DWITH_SWIG_PY=ON -DCUDNN_ROOT=/usr/ cmake .. -DWITH_DOC=OFF -DWITH_GPU=${WITH_GPU} -DWITH_SWIG_PY=ON\
-DCUDNN_ROOT=/usr/ -DWITH_STYLE_CHECK=OFF
make -j `nproc` make -j `nproc`
# because durning make install, there are several warning, so set +e, do not cause abort # because durning make install, there are several warning, so set +e, do not cause abort
make install make install
...@@ -32,8 +36,9 @@ paddle version # print version after build ...@@ -32,8 +36,9 @@ paddle version # print version after build
if [ ${WITH_DEMO} == "ON" ]; then if [ ${WITH_DEMO} == "ON" ]; then
apt-get install -y wget unzip perl python-matplotlib tar xz-utils bzip2 gzip coreutils\ apt-get install -y wget unzip perl python-matplotlib tar xz-utils bzip2 gzip coreutils\
sed grep graphviz sed grep graphviz libjpeg-dev zlib1g-dev
pip ${PIP_GENERAL_ARGS} install ${PIP_INSTALL_ARGS} BeautifulSoup docopt PyYAML pip ${PIP_GENERAL_ARGS} install ${PIP_INSTALL_ARGS} BeautifulSoup docopt \
PyYAML pillow
fi fi
if [ ${IS_DEVEL} == "OFF" ]; then # clean build packages. if [ ${IS_DEVEL} == "OFF" ]; then # clean build packages.
cd ~ cd ~
......
...@@ -46,7 +46,7 @@ setup(name="py_paddle", ...@@ -46,7 +46,7 @@ setup(name="py_paddle",
packages=['py_paddle'], packages=['py_paddle'],
include_dirs = [np.get_include(), "../"], # include numpy and paddle. include_dirs = [np.get_include(), "../"], # include numpy and paddle.
install_requires = [ install_requires = [
'numpy>=1.10.1', # The numpy is required. 'numpy>=1.8.0', # The numpy is required.
'protobuf>=2.4.1' # The paddle protobuf version 'protobuf>=2.4.1' # The paddle protobuf version
], ],
) )
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册