- Libtorch安装
(https://blog.csdn.net/qq_27278957/article/details/120014289)
在使用Libtorch部署YOLOV5的过程中,使用CPU模型成功部署,没有遇到问题,但加载GPU模型是报错,错误类型c10::NotImplementedError。
- 检查CUDA是否可用
https://blog.csdn.net/weixin_34910922/article/details/109344569
按博主说的检测出cuda不可用,在链接器命令行中添加:
/INCLUDE:?warp_size@cuda@at@@YAHXZ
CUDA成功调用:“cuda::isavailable= 1”,
重新测试加载模型,仍然报错,错误类型c10::NotImplementedError
- 最终解决方案
http://blog.17baishi.com/16743/
方法,将上述命令行改为:
/INCLUDE:"?ignore_this_library_placeholder@@YAHXZ"
模型成功加载!