PaddlePaddle in Docker Containers¶
Docker container is currently the only officially-supported way to running PaddlePaddle. This is reasonable as Docker now runs on all major operating systems including Linux, Mac OS X, and Windows. Please be aware that you will need to change Dockers settings to make full use of your hardware resource on Mac OS X and Windows.
Usage of CPU-only and GPU Images¶
For each version of PaddlePaddle, we release 2 types of Docker images: development
image and production image. Production image includes CPU-only version and a CUDA
GPU version and their no-AVX versions. We put the docker images on
dockerhub.com. You can find the
latest versions under “tags” tab at dockerhub.com.
1. development image paddlepaddle/paddle:<version>-dev
This image has packed related develop tools and runtime environment. Users and developers can use this image instead of their own local computer to accomplish development, build, releasing, document writing etc. While different version of paddle may depends on different version of libraries and tools, if you want to setup a local environment, you must pay attention to the versions. The development image contains: - gcc/clang - nvcc - Python - sphinx - woboq - sshd Many developers use servers with GPUs, they can use ssh to login to the server and run
docker exec
to enter the docker container and start their work. Also they can start a development docker image with SSHD service, so they can login to the container and start work.To run the CPU-only image as an interactive container:
docker run -it --rm paddledev/paddle:<version> /bin/bashor, we can run it as a daemon container
docker run -d -p 2202:22 -p 8888:8888 paddledev/paddle:<version>and SSH to this container using password
root
:ssh -p 2202 root@localhost
An advantage of using SSH is that we can connect to PaddlePaddle from more than one terminals. For example, one terminal running vi and another one running Python interpreter. Another advantage is that we can run the PaddlePaddle container on a remote server and SSH to it from a laptop.
- Production images, this image might have multiple variants:
- GPU/AVX:
paddlepaddle/paddle:<version>-gpu
- GPU/no-AVX:
paddlepaddle/paddle:<version>-gpu-noavx
- CPU/AVX:
paddlepaddle/paddle:<version>
- CPU/no-AVX:
paddlepaddle/paddle:<version>-noavx
Please be aware that the CPU-only and the GPU images both use the AVX instruction set, but old computers produced before 2008 do not support AVX. The following command checks if your Linux computer supports AVX:
if cat /proc/cpuinfo | grep -i avx; then echo Yes; else echo No; fi If it doesn't, we will use the non-AVX images.
Notice please don’t forget to install CUDA driver and let Docker knows about it:
export CUDA_SO="$(\ls /usr/lib64/libcuda* | xargs -I{} echo '-v {}:{}') $(\ls /usr/lib64/libnvidia* | xargs -I{} echo '-v {}:{}')" export DEVICES=$(\ls /dev/nvidia* | xargs -I{} echo '--device {}:{}') docker run ${CUDA_SO} ${DEVICES} -it paddledev/paddle:<version>-gpu
- GPU/AVX:
- Use production image to release you AI application
Suppose that we have a simple application program in
a.py
, we can test and run it using the production image:`bash docker run -it -v $PWD:/work paddle /work/a.py `
But this works only if all dependencies of
a.py
are in the production image. If this is not the case, we need to build a new Docker image from the production image and with more dependencies installs.
PaddlePaddle Book¶
The Jupyter Notebook is an open-source web application that allows you to create and share documents that contain live code, equations, visualizations and explanatory text in a single browser.
PaddlePaddle Book is an interactive Jupyter Notebook for users and developers. We already exposed port 8888 for this book. If you want to dig deeper into deep learning, PaddlePaddle Book definitely is your best choice.
We provide a packaged book image, simply issue the command:
docker run -p 8888:8888 paddlepaddle/book
Then, you would back and paste the address into the local browser:
http://localhost:8888/
That’s all. Enjoy your journey!
Development Using Docker¶
Developers can work on PaddlePaddle using Docker. This allows developers to work on different platforms – Linux, Mac OS X, and Windows – in a consistent way.
Build the Development Docker Image
git clone --recursive https://github.com/PaddlePaddle/Paddle cd Paddle docker build -t paddle:dev .
Note that by default
docker build
wouldn’t import source tree into the image and build it. If we want to do that, we need docker the development docker image and then run the following command:docker run -v $PWD:/paddle -e "WITH_GPU=OFF" -e "WITH_AVX=ON" -e "TEST=OFF" paddle:dev
Run the Development Environment
Once we got the image
paddle:dev
, we can use it to develop Paddle by mounting the local source code tree into a container that runs the image:docker run -d -p 2202:22 -p 8888:8888 -v $PWD:/paddle paddle:dev sshd
This runs a container of the development environment Docker image with the local source tree mounted to
/paddle
of the container.The above
docker run
commands actually starts an SSHD server listening on port 2202. This allows us to log into this container with:ssh root@localhost -p 2202
Usually, I run above commands on my Mac. I can also run them on a GPU server
xxx.yyy.zzz.www
and ssh from my Mac to it:my-mac$ ssh root@xxx.yyy.zzz.www -p 2202
Build and Install Using the Development Environment
Once I am in the container, I can use
paddle/scripts/docker/build.sh
to build, install, and test Paddle:/paddle/paddle/scripts/docker/build.sh
This builds everything about Paddle in
/paddle/build
. And we can run unit tests there:cd /paddle/build ctest
Documentation¶
Paddle Docker images include an HTML version of C++ source code generated using woboq code browser. This makes it easy for users to browse and understand the C++ source code.
As long as we give the Paddle Docker container a name, we can run an additional Nginx Docker container to serve the volume from the Paddle container:
docker run -d --name paddle-cpu-doc paddle:<version>
docker run -d --volumes-from paddle-cpu-doc -p 8088:80 nginx
Then we can direct our Web browser to the HTML version of source code at http://localhost:8088/paddle/