- 博客(2)
- 收藏
- 关注
RSS订阅原创 基于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
2037
原创 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
11261
8
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人