Update README.md

This commit is contained in:
Somdev Sangwan 2023-05-30 05:34:26 +05:30 committed by GitHub
parent 36e960e3b6
commit 1726a69a03
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -26,7 +26,7 @@ If you plan on doing it, you will need to install the appropriate `onnxruntime-*
#### NVIDIA
Install `cuda` and then,
```
pip install onnxruntime-gpu
pip install onnxruntime-gpu==1.15.0
```
#### AMD
Install ROCM-based torch packages from [here](https://pytorch.org/get-started/locally/) and then,