Update README.md for amd gpu
parent
532e7c05ee
commit
cc7028d3f6
11
README.md
11
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)**
|
||||
|
|
Loading…
Reference in New Issue