-
|
求问大佬，这个下载与cuda、cudnn对应版本正确的onnxruntime-gpu，设置use_cuda=True,就可以再GPU上跑了吗? |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment 4 replies
-
|
emmm,cudnn我没试过，可能不行，cuda的应该可以 |
Beta Was this translation helpful? Give feedback.
-
|
大佬可以简答说下在cuda上怎么用吗 |
Beta Was this translation helpful? Give feedback.
-
|
就是你说的这样，目前直接支持使用gpu执行，但是性能好像没有提升，主要是避免cpu突然增长的问题，但是也不慢，大概100ms以内 |
Beta Was this translation helpful? Give feedback.
-
|
完整示例里面有哈，就是一个参数就可以了，但是需要安装onnxruntime-gpu |
Beta Was this translation helpful? Give feedback.
-
好的，谢谢佬 |
Beta Was this translation helpful? Give feedback.
完整示例里面有哈，就是一个参数就可以了，但是需要安装onnxruntime-gpu