docker编译paddlepaddle记录

环境

CUDA 10.0
cudann 7
paddlepaddle-realse-1.8.0

步骤

下载源码

git clone https://github.com/PaddlePaddle/Paddle.git

创建docker容器

cd Paddle
nvidia-docker run --name paddle-test -v $PWD:/paddle --network=host -it hub.baidubce.com/paddlepaddle/paddle:latest-gpu-cuda10.0-cudnn7-dev /bin/bash

进入docker的paddle目录下

cd paddle
git branch -a # 查看所有分支
git checkout 分支名字

准备编译

mkdir -p /paddle/build && cd /paddle/build
pip install protobuf -i https://mirror.baidu.com/pypi/simple

编译

cmake .. -DPY_VERSION=3.5 -DWITH_GPU=ON -DWITH_TESTING=OFF -DCMAKE_BUILD_TYPE=Release
make -j$(nproc) # 多核编译

编译成功后进入/paddle/build/python/dist目录下找到生成的.whl包： cd /paddle/build/python/dist

安装

pip install -U（whl包的名字）

参考

https://www.paddlepaddle.org.cn/documentation/docs/zh/install/compile/compile_Ubuntu.html