Replies: 1 comment
-
是的,但是要确保onnxruntime-gpu可以正确调用GPU,onnxruntime-gpu对cuda和cudnn有严格版本限制,需要去官方文档看一下了。 |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
求问大佬,这个下载与cuda、cudnn对应版本正确的onnxruntime-gpu,设置use_cuda=True,就可以再GPU上跑了吗?
Beta Was this translation helpful? Give feedback.
All reactions