所以我有一个预训练的pytorch模型，以防万一，我已经保存了模型和参数，但是我需要在C或C ++代码中使用此模型，有人知道我该怎么做吗？
非常感谢。
答案 0 :(得分：0)
尝试pytorch 1.0，其中包括一个c ++库。该模型可以导出为割炬脚本格式并以C ++加载，请在此处https://pytorch.org/tutorials/advanced/cpp_export.html
答案 1 :(得分：0)
如果您正在使用Ubuntu，请尝试使用ppa:nnstreamer/ppa的PPA中的pytorch软件包。 在此Ubuntu pytorch软件包中，有头文件和pkgconfig（.pc）文件，可帮助使用pytorch构建C / C ++程序。 如果您想要更简单的方法来从C / C ++程序调用pytorch / caffe2模型，则可以使用nnstreamer的C / API或也可以使用带有GStreamer API的nnstreamer。
要安装nnstreamer或C友好的pytorch（或tensorflow），请通过以下方式使用PPA：
$ sudo add-apt-repository ppa:nnstreamer/ppa
$ sudo apt-get update
$ suto apt-get install ${whateveryouwant}
$ {whateveryouwant}的候选人：nnstreamer，nnstreamer-dev，nnstreamer-pytorch
如果您想直接使用C代码直接访问pytorch而不使用nnstreamer，则可以在以下位置参考nnstreamer的pytorch-wrapper：https://github.com/nnsuite/nnstreamer/blob/master/ext/nnstreamer/tensor_filter/tensor_filter_pytorch_core.cc