parent
3ca5df0f47
commit
0276f15a45
@ -0,0 +1 @@
|
||||
dist
|
@ -0,0 +1,5 @@
|
||||
FROM paddledev/paddle:gpu-latest
|
||||
MAINTAINER PaddlePaddle Dev Team <paddle-dev@baidu.com>
|
||||
COPY build.sh /root/
|
||||
CMD cd /root/ && bash build.sh
|
||||
|
@ -0,0 +1,37 @@
|
||||
#!/bin/bash
|
||||
set -e
|
||||
apt-get install -y dh-make
|
||||
cd ~
|
||||
mkdir -p ~/dist/gpu
|
||||
mkdir -p ~/dist/cpu
|
||||
mkdir -p ~/dist/cpu-noavx
|
||||
mkdir -p ~/dist/gpu-noavx
|
||||
git clone https://github.com/baidu/Paddle.git paddle
|
||||
cd paddle
|
||||
mkdir build
|
||||
cd build
|
||||
cmake .. -DWITH_GPU=OFF -DWITH_SWIG_PY=ON -DWITH_AVX=ON
|
||||
make -j `nproc`
|
||||
cpack -D CPACK_GENERATOR='DEB' ..
|
||||
mv *.deb ~/dist/cpu
|
||||
|
||||
rm -rf *
|
||||
cmake .. -DWITH_GPU=ON -DWITH_SWIG_PY=ON -DWITH_AVX=ON -DCUDNN_ROOT=/usr/
|
||||
make -j `nproc`
|
||||
cpack -D CPACK_GENERATOR='DEB' ..
|
||||
mv *.deb ~/dist/gpu
|
||||
|
||||
|
||||
rm -rf *
|
||||
cmake .. -DWITH_GPU=OFF -DWITH_SWIG_PY=ON -DWITH_AVX=OFF
|
||||
make -j `nproc`
|
||||
cpack -D CPACK_GENERATOR='DEB' ..
|
||||
mv *.deb ~/dist/cpu-noavx
|
||||
|
||||
rm -rf *
|
||||
cmake .. -DWITH_GPU=ON -DWITH_SWIG_PY=ON -DWITH_AVX=OFF -DCUDNN_ROOT=/usr/
|
||||
make -j `nproc`
|
||||
cpack -D CPACK_GENERATOR='DEB' ..
|
||||
mv *.deb ~/dist/gpu-noavx
|
||||
|
||||
|
@ -0,0 +1,8 @@
|
||||
#!/bin/bash
|
||||
set -e
|
||||
docker build -t build_paddle_deb .
|
||||
rm -rf dist
|
||||
mkdir -p dist
|
||||
docker run -v$PWD/dist:/root/dist --name tmp_build_deb_container build_paddle_deb
|
||||
docker rm tmp_build_deb_container
|
||||
docker rmi build_paddle_deb
|
Loading…
Reference in new issue