Update README.md
This commit is contained in:
parent
36e960e3b6
commit
1726a69a03
@ -26,7 +26,7 @@ If you plan on doing it, you will need to install the appropriate `onnxruntime-*
|
|||||||
#### NVIDIA
|
#### NVIDIA
|
||||||
Install `cuda` and then,
|
Install `cuda` and then,
|
||||||
```
|
```
|
||||||
pip install onnxruntime-gpu
|
pip install onnxruntime-gpu==1.15.0
|
||||||
```
|
```
|
||||||
#### AMD
|
#### AMD
|
||||||
Install ROCM-based torch packages from [here](https://pytorch.org/get-started/locally/) and then,
|
Install ROCM-based torch packages from [here](https://pytorch.org/get-started/locally/) and then,
|
||||||
|
Loading…
Reference in New Issue
Block a user