Update README.md

This commit is contained in:
SWHL authored and GitHub committed 2022-05-03 10:40:11 +08:00
1 parent d983c693c8
commit 2fc0096e75
1 file changed
+3
+3
View File
@@ -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>'