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
|
pip install onnxruntime-gpu==1.16.3
|
||||||
```
|
```
|
||||||
|
|
||||||
|
**ROCM Execution Provider (AMD GPU)**
|
||||||
|
```bash
|
||||||
|
pip uninstall onnxruntime onnxruntime-gpu
|
||||||
|
pip install onnxruntime-directml
|
||||||
|
```
|
||||||
|
|
||||||
3. Usage:
|
3. Usage:
|
||||||
|
|
||||||
```bash
|
```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)**
|
**CoreML Execution Provider (Apple Silicon)**
|
||||||
|
|
Loading…
Reference in New Issue