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>'