From 8c073f9b5f93c35c230cdde3db1cde3e6fa7b598 Mon Sep 17 00:00:00 2001 From: Maanav Dalal Date: Mon, 8 Jul 2024 13:27:13 -0700 Subject: [PATCH] Added requested install instructions to ORT ROCm Python. (#21124) To close: #21036 --- docs/install/index.md | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/docs/install/index.md b/docs/install/index.md index 061813b827aa3..a7839874857b1 100644 --- a/docs/install/index.md +++ b/docs/install/index.md @@ -60,6 +60,12 @@ 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 from source on Linux, follow the instructions [here](https://onnxruntime.ai/docs/build/eps.html#amd-rocm). Alternatively, each major ORT release has a corresponding C/C++ ROCm package, found [here](https://github.com/microsoft/onnxruntime/releases/). + + ### Install ONNX to export the model ```bash