环境
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