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