Skip to content

GitLab

Unverified Commit 692e1956 authored by Hubert Lu's avatar Hubert Lu Committed by GitHub
Browse files

Update README.md

parent cc92a4b4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
README.md
View file @ 692e1956
...@@ -136,6 +136,7 @@ python setup.py install ...@@ -136,6 +136,7 @@ python setup.py install
``` ```
python setup.py install --cpp_ext --cuda_ext python setup.py install --cpp_ext --cuda_ext
``` ```
Note that using --cuda_ext flag to install Apex will also enable all the extensions supported on ROCm including "--distributed_adam", "--distributed_lamb", "--bnp", "--xentropy", "--deprecated_fused_adam", "--deprecated_fused_lamb", and "--fast_multihead_attn".
### To install Apex on ROCm using ninja and without cloning the source ### To install Apex on ROCm using ninja and without cloning the source
``` ```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment