![](https://img-blog.csdnimg.cn/20201014180756738.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
AI部署
mikeyNa
这个作者很懒,什么都没留下…
展开
-
libtorch在windows上的使用及采坑
libtorch在windows下的使用原创 2022-09-07 15:38:07 · 990 阅读 · 0 评论 -
OnnxRuntime C++不能使用CUDA的问题
问题描述:当使用onnxruntime进行推理的时候发现使用的是cpu,而不是GPU解决办法:有可能是头文件和设置的问题step1:头文件添加:#include <onnxruntime_c_api.h>#include <cuda_provider_factory.h>step2:session设置:Ort::SessionOptions session_options;session_options.SetInterOpNumThreads(1原创 2022-05-31 16:40:36 · 2865 阅读 · 0 评论