mirror of
https://github.com/PaddlePaddle/PaddleClas.git
synced 2025-06-03 21:55:06 +08:00
37 lines
744 B
Markdown
37 lines
744 B
Markdown
# PaddleClas 下静态图benchmark模型执行说明
|
||
|
||
静态图benchmark测试脚本说明
|
||
|
||
# 目录说明
|
||
# Docker 运行环境
|
||
docker image: registry.baidubce.com/paddlepaddle/paddle:latest-dev-cuda11.2-cudnn8-gcc82
|
||
paddle = 2.2.2
|
||
python = 3.7
|
||
# 运行benchmark测试步骤
|
||
|
||
```shell
|
||
git clone https://github.com/PaddlePaddle/PaddleClas.git
|
||
cd PaddleClas
|
||
```
|
||
|
||
# 准备数据
|
||
|
||
```shell
|
||
bash test_tipc/static/${model_item}/benchmark_common/prepare.sh
|
||
```
|
||
|
||
# 运行模型
|
||
## 单卡（自动运行打开Profiling）
|
||
|
||
```shell
|
||
export CUDA_VISIBLE_DEVICES=0
|
||
bash test_tipc/static/${model_item}/N1C1/${shell_name}.sh
|
||
```
|
||
|
||
## 多卡
|
||
|
||
```shell
|
||
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
|
||
bash test_tipc/static/${model_item}/N1C8/${shell_name}.sh
|
||
```
|