Skip to content

Commit

Permalink
update readme
Browse files Browse the repository at this point in the history
  • Loading branch information
zhouyu5 committed Jun 13, 2024
1 parent f34be93 commit 4d99374
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions benchmark/multi_gpu/training/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -44,10 +44,10 @@ pip install torch==2.1.0.post2 intel-extension-for-pytorch==2.1.30+xpu --extra-i

### Running benchmark

Run benchmark, e.g. assuming you have `n` XPUs:
This [guide](https://intel.github.io/intel-extension-for-pytorch/xpu/latest/tutorials/features/DDP.html) is helpful for you to lauch DDP training on intel GPU.

To Run benchmark, e.g. assuming you have `n` XPUs:

```
mpirun -n <n> python training_benchmark_xpu.py --dataset ogbn-products --model edge_cnn --num-epochs 3
mpirun -np <n> python training_benchmark_xpu.py --dataset ogbn-products --model edge_cnn --num-epochs 3
```

This [guide](https://intel.github.io/intel-extension-for-pytorch/xpu/latest/tutorials/features/DDP.html) is also helpful for you to lauch DDP training on intel GPU.

0 comments on commit 4d99374

Please sign in to comment.