Update README.md for amd gpu

pull/1175/head
Pawan 2025-05-03 18:30:37 +05:30 committed by GitHub
parent 532e7c05ee
commit cc7028d3f6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 1 deletions

View File

@ -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)**