![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
libtorch
文章平均质量分 72
不会飞的鹰08
不搞空中楼阁,只做落地项目
展开
-
libtorch-win10环境配置
版本问题pytorch 版本和libtorch版本一致,原有的环境pytorch 1.6.0,这里采用pytorch 1.7.0win10 cuda 版本 11.0查看自己的cuda版本nvcc -V 复制一份当前环境conda create -n sotcuda11 --clone sotconda activate sotcuda11安装pytorch 1.7.0,本地是cuda11.0.3https://pytorch.org/get-started/previous原创 2021-09-01 16:39:23 · 1254 阅读 · 0 评论 -
libtorch-模型导出
概念区分jit.script 与 jit.trace:[PyTorch] jit.script 与 jit.trace简单区分:torch.jit.script 转化不了的模型可以试试torch.jit.trace推荐使用的是jit.traceyolov5中的导出模型使用的便是torch.jit.tracehttps://github.com/ultralytics/yolov5/blob/8b18b66304317276f4bfc7cc7741bd535dc5fa7a/export.py原创 2021-09-01 12:42:40 · 554 阅读 · 0 评论