From cc7028d3f6e8e6f5fef0826ef2133002ec338c6a Mon Sep 17 00:00:00 2001 From: Pawan <75882609+iitzrao@users.noreply.github.com> Date: Sat, 3 May 2025 18:30:37 +0530 Subject: [PATCH] Update README.md for amd gpu --- README.md | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 6dfee6b..32e95d5 100644 --- a/README.md +++ b/README.md @@ -195,10 +195,19 @@ pip uninstall onnxruntime onnxruntime-gpu pip install onnxruntime-gpu==1.16.3 ``` +**ROCM Execution Provider (AMD GPU)** +```bash +pip uninstall onnxruntime onnxruntime-gpu +pip install onnxruntime-directml +``` + 3. Usage: ```bash -python run.py --execution-provider cuda +python run.py --execution-provider cuda //for nvidia +``` +```bash +python run.py --execution-provider dml //for amd ``` **CoreML Execution Provider (Apple Silicon)**