未验证 提交 afd11a40 编写于 作者: S Snow 提交者: GitHub

Update README.md (#76)

* Update README.md

* Update README.md
上级 8d47db1b
......@@ -54,6 +54,34 @@ cd models/PaddleCV/image_classification
python3 -m pip install paddlepaddle-gpu==1.8.3.post107 -i https://mirror.baidu.com/pypi/simple
```
## DALI
1. 本测试使用 conda 维护测试环境, 进入 paddle 的 conda 用例, 下载 DALI_extra 源码
```
conda activate paddle
git clone https://github.com/NVIDIA/DALI_extra.git
```
2. 使用 conda 环境内的 python 安装 nightly DALI
```
/home/leinao/anaconda3/envs/paddle/bin/python -m pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/nightly nvidia-dali-nightly-cuda100
```
3. 设置 `export FLAGS_fraction_of_gpu_memory_to_use=0.8`
```
export FLAGS_fraction_of_gpu_memory_to_use=0.8
```
否则可能会报错
```
AssertionError: Please leave enough GPU memory for DALI workspace, e.g., by setting `export FLAGS_fraction_of_gpu_memory_to_use=0.8
```
## NCCL
paddle的分布式训练底层依赖NCCL库,需要从[NVIDIA-NCCL官网下载](https://developer.nvidia.com/nccl/nccl-download)并安装和操作系统、CUDA版本适配的NCCL。
......@@ -286,4 +314,4 @@ README展示的是extract_paddle_logs.py的计算结果。
- [resnet50-fp32.zip](https://oneflow-public.oss-cn-beijing.aliyuncs.com/DLPerf/logs/PaddlePaddle/resnet50/resnet50_fp32.zip)
- [resnet50-fp16.zip](https://oneflow-public.oss-cn-beijing.aliyuncs.com/DLPerf/logs/PaddlePaddle/resnet50/resnet50_fp16.zip)
\ No newline at end of file
- [resnet50-fp16.zip](https://oneflow-public.oss-cn-beijing.aliyuncs.com/DLPerf/logs/PaddlePaddle/resnet50/resnet50_fp16.zip)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册