From 2fc0096e752a51ae9e25388380a287cbe6f80cea Mon Sep 17 00:00:00 2001
From: SWHL <liekkaskono@163.com>
Date: Tue, 3 May 2022 10:40:11 +0800
Subject: [PATCH] Update README.md

---
 python/onnxruntime_infer/README.md | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/python/onnxruntime_infer/README.md b/python/onnxruntime_infer/README.md
index 5b84f3f..b6ccb53 100644
--- a/python/onnxruntime_infer/README.md
+++ b/python/onnxruntime_infer/README.md
@@ -110,6 +110,9 @@
 2. 推理代码中，加载onnx模型部分，用以下对应语言代码替换即可,详细参见：[官方教程](https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html)
    - python版本
       ```python
+      # 根据机器配置，安装对应版本的onnxruntime-gpu
+      # pip install onnxruntime-gpu==xxxx
+            
       import onnxruntime as ort
 
       model_path = '<path to model>'