PaddleClas/docs/zh_CN/tutorials/install.md

1.9 KiB
Raw Blame History

安装说明


1.简介

本章将介绍如何安装PaddleClas及其依赖项. 有关模型库的基本信息请参考README

2.安装PaddlePaddle

运行PaddleClas需要PaddlePaddle Fluid v1.7或更高版本。

pip安装最新GPU版本PaddlePaddle

pip install paddlepaddle-gpu --upgrade

或是从源码安装PaddlePaddle具体参照安装文档中的说明进行操作。

使用以下命令验证

import paddle.fluid as fluid
fluid.install_check.run_check()

查看PaddlePaddle版本

python -c "import paddle; print(paddle.__version__)"

注意:

  • 从源码编译的PaddlePaddle版本号为0.0.0请确保使用了Fluid v1.7之后的源码编译。
  • PaddleClas基于PaddlePaddle高性能的分布式训练能力若您从源码编译请确保打开编译选项WITH_DISTRIBUTE=ON。具体编译选项参考编译选项表

运行环境需求:

  • Python2官方已不提供更新维护或Python3 (windows系统仅支持Python3)
  • CUDA >= 9.0
  • cuDNN >= 5.0
  • nccl >= 2.1.2

3.安装PaddleClas

克隆PaddleClas模型库

cd path_to_clone_PaddleClas
git clone https://github.com/PaddlePaddle/PaddleClas.git

安装Python依赖库

Python依赖库在requirements.txt中给出,可通过如下命令安装:

pip install --upgrade -r requirements.txt

4.数据集和预训练模型

PaddleClas加载PaddleClas/dataset/中数据进行训练,请参照数据文档进行准备。 PaddleClas提供丰富的预训练模型请参照数据文档进行准备。

5.开始使用

请参照开始使用文档