Linux

PaddlePaddle安装

2017-11-26  本文已影响4768人  sweetsky0901

欢迎访问PaddlePaddle的社区官网
官方最新的安装文档 合适更专业的同学阅读
你可以使用提供的二进制包安装,也可以使用源代码编译来安装PaddlePaddle。
当然,为了方便也可以直接使用Docker镜像。
本文重点讲解如何使用二进制安装和源码安装

二进制安装

对系统的要求

在Linux和Mac下最简单的安装方式,是使用pip安装.
如果在安装过程中遇到错误,请查阅常见问题

Ubuntu/Linux/ MacOS X

仅使用CPU的版本

$ pip install paddlepaddle

开启GPU支持的版本(安装该版本的前提是已经安装了CUDA sdk)

$ pip install paddlepaddle_gpu


源码安装

准备工作

curl -SSL https://cmake.org/files/v3.4/cmake-3.4.1.tar.gz | tar -xz &&
cd cmake-3.4.1 && ./bootstrap && make -j4 && sudo make install &&
cd .. && rm -rf cmake-3.4.1

其他编译依赖(不同的操作系统,详见)

下载源码并编译安装

下载源代码

$ git clone https://github.com/PaddlePaddle/Paddle.gitpaddle
$ cd paddle

生成makefile

$ mkdir build
$ cd build
$ cmake ..

编译安装

$ make -j nproc
$ sudo make install

cmake选项说明:

cmake .. -DCMAKE_INSTALL_PREFIX=$PADDLE_INSTALL_PATH,

cmake .. -DWITH_MKLML=OFF -DWITH_MKLDNN=OFF -DWITH_GPU=ON -DWITH_C_API=OFF -DWITH_TESTING=ON

DWITH_GPU:表示是否开启GPU
DWITH_TESTING:表示是否开启单测

cmake .. -DPYTHON_EXECUTABLE=<exc_path> -DPYTHON_LIBRARY=<lib_path> -DPYTHON_INCLUDE_DIR=<inc_path>

常见问题

cuda和cudnn的版本以及GPU卡的型号

paddlepaddle的pip安装的gpu的包,是基于cuda8和cudnn5来编译的,所以请确保你的环境中的cuda和cudnn的版本与我们是一致的,否则运行的时候会报错。
paddlepaddle在计划发行更多的pip可以直接安装的版本,以支持不同版本的cuda和cudnn
即使使用源码编译安装,paddlepaddle当前也依旧不支持cuda9,但是已经在计划中了。

cuda和cudnn的下载安装

GPU 特性只支持 NVidia Compute Capability >= 3.0 的显卡
cuda8的下载地址是:cuda8
cudnn5的下载地址是:cudnn5
请下载【Download cuDNN v5 (May 27, 2016), for CUDA 8.0】
你还需要设置 LD_LIBRARY_PATH 环境变量。 可以考虑将下面的命令 添加到 ~/.bash_profile 文件中, 这样每次登陆后自动生效. 注意, 下面的命令 假定 CUDA 安装目录为 /usr/local/cuda:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

源码编译说明

因为当前mkl还有一些不稳定,建议跳过。
如果你在编译cpu的版本,建议cmake内容:
cmake .. -DWITH_MKLML=OFF -DWITH_MKLDNN=OFF -DWITH_GPU=OFF -DWITH_C_API=OFF -DWITH_TESTING=OFF
如果你在编译GPU版本,建议cmake内容
cmake .. -DWITH_MKLML=OFF -DWITH_MKLDNN=OFF -DWITH_GPU=ON -DWITH_C_API=OFF -DWITH_TESTING=OFF
其他更详细的选项说明,参照官网

上一篇下一篇

猜你喜欢

热点阅读