Added requested install instructions to ORT ROCm Python.

microsoft · Jun 20, 2024 · 844e433 · 844e433
1 parent 77df36b
commit 844e433
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/docs/install/index.md b/docs/install/index.md
@@ -60,6 +60,16 @@ For Cuda 12.x, please use the following instructions to install from [ORT Azure
 pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
 ```
 
+#### Install ONNX Runtime GPU (ROCm)
+For ROCm, please follow instructions to install it at the [AMD ROCm install docs](https://rocm.docs.amd.com/projects/install-on-linux/en/docs-6.0.0/). The ROCm execution provider for ONNX Runtime is built and tested with ROCm 6.0.0
+
+To build on Linux, use the following instructions: 
+```bash
+./build.sh --config <Release|Debug|RelWithDebInfo> --use_rocm --rocm_home
+```
+
+See more information about the ROCm Execution Provider [here](https://onnxruntime.ai/docs/execution-providers/ROCm-ExecutionProvider.html).
+
 ### Install ONNX to export the model
 
 ```bash