Pytorch C++ libtorch的使用方法

最新推荐文章于 2024-07-31 17:28:45 发布

chencision

最新推荐文章于 2024-07-31 17:28:45 发布

阅读量2w

点赞数 9

分类专栏：分享文章标签： pytorch 深度学习

本文链接：https://blog.csdn.net/baidu_34595620/article/details/101532239

版权

Pytorch C++ libtorch

pytorch C++ libtorch的，版本方面可以自己进行源码编译（有很多依赖），也可以从github上下载已经编译好的版本，官方使用教程给的libtorch编译时g++版本较低，不支持CXX11所以，可以下载支持CXX11版本的libtorch

具体使用

对于pytorch的模型使用官方已有很明确的教程，可以放心使用，其中从pth->pt时，可能会遇到关于hooks的问题，我找到的方法首先

# pth->pt pt用来给C++使用
import torch
from torch import nn
from collections import OrderedDict
from collections import namedtuple
def remove_hooks(model):
    model._backward_hooks = OrderedDict()
    model._forward_hooks = OrderedDict()
    model._forward_pre_hooks = OrderedDict()
    for child in model.children():
        remove_hooks(child)
device = torch.device('cpu')
model=torch.