![](https://img-blog.csdnimg.cn/20201014180756919.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
芯光智能
这个作者很懒,什么都没留下…
展开
-
基于PyTorch的C++API运行模型进行图像分类
一个简单示例如下: import torch #import torchvision class MyCell(torch.nn.Module): def __init__(self): super(MyCell, self).__init__() def forward(self, x, h): new_h = torch.tanh(x + h) return new_h, new_h my_cell = MyCell() x =原创 2020-12-01 15:55:58 · 565 阅读 · 0 评论 -
Pytorch 环境搭建C++
1,阅读Pytorch C++官网文档,在虚拟机上安装Ubuntu16.04的Linux操作系统搭建环境,并使用makefile进行编译,代码简洁易读。 sudo apt-get update sudo apt-get install vim make cmake gcc g++ libnss3 tree git openssh-server openssh-client 2,下载pytorch c++ cpu 库文件,并解压,注意此目录路径,cmake编译需要 unzip libtorch-s原创 2020-11-30 14:56:54 · 1093 阅读 · 0 评论