PaddleClas/docs/zh_CN/tutorials/install.md

2.5 KiB
Raw Blame History

安装说明


一、简介

本章将介绍如何安装PaddleClas及其依赖项。

二、安装PaddlePaddle

运行PaddleClas需要PaddlePaddle 2.0或更高版本。请参照安装文档中的说明进行操作。

通过pip安装PaddlePaddle

如果已经安装好了cuda、cudnn、nccl或者安装好了docker、nvidia-docker运行环境可以pip安装最新GPU版本PaddlePaddle

pip3 install paddlepaddle-gpu --upgrade -i https://mirror.baidu.com/pypi/simple

如果希望在CPU环境中使用PaddlePaddle可以运行下面的命令安装PaddlePaddle。

pip3 install paddlepaddle --upgrade -i https://mirror.baidu.com/pypi/simple

源码编译PaddlePaddle

您也可以从源码编译安装PaddlePaddle请参照安装文档中的说明进行操作。

使用以下命令可以验证PaddlePaddle是否安装成功。

import paddle
paddle.utils.run_check()

查看PaddlePaddle版本的命令如下

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

注意:

  • 从源码编译的PaddlePaddle版本号为0.0.0请确保使用了PaddlePaddle 2.0及之后的源码编译。
  • PaddleClas基于PaddlePaddle高性能的分布式训练能力若您从源码编译请确保打开编译选项WITH_DISTRIBUTE=ON。具体编译选项参考编译选项表
  • 在docker中运行时为保证docker容器有足够的共享内存用于Paddle的数据读取加速在创建docker容器时请设置参数--shm_size=8g,条件允许的话可以设置为更大的值。

运行环境需求:

  • Python3
  • CUDA >= 9.0
  • cuDNN >= 7.6.4
  • nccl >= 2.1.2

三、安装PaddleClas

克隆PaddleClas模型库

git clone https://github.com/PaddlePaddle/PaddleClas.git -b develop

如果从github上网速太慢可以从gitee下载下载命令如下

git clone https://gitee.com/paddlepaddle/PaddleClas.git -b develop

安装Python依赖库

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

pip3 install --upgrade -r requirements.txt -i https://mirror.baidu.com/pypi/simple

visualdl可能出现安装失败请尝试

pip3 install --upgrade visualdl -i https://mirror.baidu.com/pypi/simple

此外visualdl目前只支持在python3下运行因此如果希望使用visualdl需要使用python3。