摘要:
一、基本使用方法 二、调用简单语句 三、调用函数 四、调用类 五、调用SSD目标检测算法 六、遇到的错误 最近训练一个3D分割的模型,需要将其结合到项目中,由于项目是C开发,而这边python训练好的模型尝试了ONNX、libtorch等转换C也没有成功,因此考虑采用C++直接调用python代码, 阅读全文
摘要:
一、基本使用方法 二、调用简单语句 三、调用函数 四、调用类 五、调用SSD目标检测算法 六、遇到的错误 三、调用函数 3.1 无参 -CMakeLists.txt cmake_minimum_required(VERSION 3.9) project(say_hello) set(SDK_VERS 阅读全文
摘要:
参考链接 如何计算模型以及中间变量的显存占用大小: https://oldpan.me/archives/how-to-calculate-gpu-memory 如何在Pytorch中精细化利用显存: https://oldpan.me/archives/how-to-use-memory-pyto 阅读全文