阅读量:0
Pytorch模型转换Caffe模型踩坑指南,代码使用的是Github上的工程,地址:
https://github.com/longcw/pytorch2caffe
操作环境:ubuntu = 14.04 miniconda 3 caffe pytorch = 0.2.0 torchvision = 0.1.8 python = 2.7
环境配置:
第一步 : 在miniconda创建一个虚拟环境pytorch2caffe : conda create -n pytorch2caffe python=2.7
第二步 : 激活虚拟环境 source activate pytorch2caffe
第三步 : 在该虚拟环境下安装对应版本的pytorch和torchvision : conda install pytorch=0.2.0 torchvision=0.1.8 安装完成后conda list看一看有没有安装成功对应的版本
第四步 : 在该虚拟环境下编译caffe,官网指南链接 :
http://caffe.berkeleyvision.org/install_apt.html
第五步 : 编译配置caffe的python接口pycaffe,操作指南链接 :
https://www.cnblogs.com/lyyang/p/6573846.html
配置环境的时候因Git