From 1726a69a03596bf9f771bd162a04f6c41b36f484 Mon Sep 17 00:00:00 2001 From: Somdev Sangwan Date: Tue, 30 May 2023 05:34:26 +0530 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index ad7218f..d345906 100644 --- a/README.md +++ b/README.md @@ -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,