模型转换工具合集与 pytorch2caffe

模型转换工具

github地址: https://github.com/ysh329/deep-learning-model-convertor
模型转换的部分截图如下:
1
海思自己的在线转换工具应该是使用了微软的MMdnn(https://github.com/Microsoft/MMdnn

pytorch2caffe

尝试一(失败)

网上很多人都用的是https://github.com/longcw/pytorch2caffe
可惜pytorch版本要求低于0.4,实在是太老了,模型都加载不起来。
虽然最后失败了,但是我也把大致步骤写出来,给大家一个参考。

1.使用源码安装pytorch0.2 (实在是找不到0.2–CPU版本的安装包)

2.import torch 出现如下问题

ImportError: /usr/local/share/anaconda2/envs/onnx_env/lib/python2.7/site-packages/torch/_C.so: undefined symbol: _ZN3MPI8Datatype4FreeEv

https://github.com/pytorch/pytorch/issues/2680里搜索,有人说有一个临时的解决办法是LD_PRELOAD=/usr/lib/libmpi_cxx.so python

这时候就需要源码编译 openmpi,编译前记得加上选项--enable-mpi-cxx,这样才会有 libmpi_cxx.so 文件。

最后打开python
LD_PRELOAD=/usr/local/openmpi/lib/libmpi_cxx.so python

加载pytorch

猜你喜欢

转载自blog.csdn.net/qq_40421682/article/details/107034247
今日推荐