Update README.md

This commit is contained in:
Somdev Sangwan 2023-05-30 05:34:26 +05:30
parent a1b38b9eb5
commit d7abe79f59

View File

@ -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,