pytorch
ZHIKELIN
这个作者很懒,什么都没留下…
展开
-
使用pytorch输出onnx模型并转换成openvino的IR模型
经过好几天的试错,终于成功转换出模型了,出错主要问题是各种包版本不一致的问题。安装pytorch/torchvision安装环境和包版本:Windows 10torch:1.5.1;torchvision:0.6.1;python:3.7.6onnx:1.7.0openVino:Anaconda3-2020.02-Windows-x86_64.exetensorflow>=1.2.0,<2.0.0mxnet>=1.0.0,<=1.3.1networkx=2.0原创 2020-07-13 17:20:44 · 3481 阅读 · 6 评论 -
pytorch模型转换输出为onnx过程
环境列表:windows 10anaconda3;pytorch 1.5.1cuda 10.1python 3.7.0onnx 1.7环境安装过程1. 安装anaconda3我安装的是个人版,下载地址:anaconda3个人版下载页面安装过程很简单,双击打开下载的exe文件就开始安装了,基本都是按默认选项安装,若不想使用默认安装C盘目录,可以按照个人需要选择其他路径,本人安装在D盘;2. 安装pytorch在anaconda环境的安装过程也是很简单,pytorch官方页面有安装的命原创 2020-07-08 14:45:14 · 4284 阅读 · 0 评论