caffe2
lshiwjx
哈?
展开
-
gcc4.8+Caffe2(0.8)+onnx(1.2)+onnx-caffe2在anaconda中的安装,不需要root,源码编译.
caffe2的cuda8预编译是gcc5以上的. gcc5以下的只有cuda9和cpu版本对于gcc4.8和cuda8的caffe2没有预编译. 源码安装caffe2git clone --recursive https://github.com/pytorch/pytorch.gitcd pytorchgit submodule update --initCONDA_INST...原创 2018-08-18 15:02:26 · 1119 阅读 · 0 评论 -
caff2 与 pytorch 模型的转换--onnx
onnx和caffe2的安装参考另一篇博客caffe2新版本中添加了onnx-caffe2,所以无需再安装Pytorch to ONNXimport torchfrom pnasnet import PNASNet5Largefrom torch.autograd import Variablemodel = PNASNet5Large(50)m = torch.load(...原创 2018-08-18 15:08:16 · 3683 阅读 · 4 评论