- 博客(2)
- 收藏
- 关注
原创 基于Pytorch语义分割模型的C++部署教程,CPU版本
基于Pytorch语义分割模型的C++部署教程,CPU版本1.pth权重文件转pt权重文件两种pth保存的方式模型转换2.C++ libtorch的下载与环境配置libtorch下载Libtorch C++环境的配置3.C++ 下pt文件的测试refer1.pth权重文件转pt权重文件两种pth保存的方式Pytorch保存模型权重文件有两种方式:1.保存整个模型及模型参数的方式:torch.save(self_model, "unet.pth")2.保存模型参数方式:torch.save(s
2021-12-26 23:36:59 2309
原创 CUDA10.2下安装tensorflow-gpu2.0.0版本注意事项
CUDA10.2下安装tensorflow-gpu2.0.0版本注意事项最近新入手了一台i5-10200U的CPU,有一个极为鸡肋的MX250显卡(2G),且经得起造的ThinkPad T490笔记本电脑,为将来的移动办公做准备。今天突然心血来潮想试试它能不能用cuda的加速,去了NVIDIA的官网查询后,发现这块鸡肋N卡还真的支持cuda加速,就这样花了大半天的时间将cuda安装好后,并进行了一个CNN的网络训练。一、安装准备需要在你的NVIDIA控制面板→右下角系统信息→组件,去查看你的NVCUD
2020-05-31 13:01:52 11529 8
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人