![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
PyTorch
torch
六月的翅膀
人生本是无根草,醉了何必问家乡
展开
-
显卡驱动安装查询
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\extras\demo_suite安装好驱动后可以到这个目录下命令行中执行这两个exe文件可以查看对应的信息也可以nvidia-smi查看显卡信息原创 2022-05-07 10:58:49 · 983 阅读 · 0 评论 -
C++调pytorch模型的全过程记录
前面已经记录过了,流程就是这么个流程:配置libtorch --->python训练的模型怎么在C++使用?_己亥谷雨-CSDN博客 pytorch模型转化 编写C++调用程序这里就来记录一下模型转化和C++调用程序。关于模型的训练就不多说了,不会训练模型还扯什么调用。好在之前记录过一点:训练模型PyTorch:VGG16简单入门版_己亥谷雨-CSDN博客 保存测试模型PyTorch:保存/加载训练好的模型测试_己亥谷雨-CSDN博客_pytorch训练好的模型去检测==..原创 2021-10-30 02:06:41 · 2630 阅读 · 3 评论 -
python训练的模型怎么在C++使用?
假设我现在已经训练好了一个模型,效果挺好,想用C++调。流程就是这么个流程:配置libtorch pytorch模型转化 编写C++调用程序这里就先看一下第一步。怎么配置libtorch。首先去官网下载PyTorch https://pytorch.org/解压后是这个样子使用方法呢和一般的库一样包含目录两个库目录一个链接器->输入->附加依赖项,这里刚开始没写,一堆错误。干脆写全了就是libtorch\lib目录下的所有lib文件名...原创 2021-10-28 16:17:05 · 7081 阅读 · 3 评论 -
PyTorch画出数据集中的图像-CIFAR10
import torchimport torch.nn as nnfrom torch import optimfrom torch.autograd import Variablefrom torch.utils.data import DataLoaderfrom torchvision import transformsfrom torchvision import datasetsfrom tqdm import tqdmimport matplotlib.pyplot as pl.原创 2021-08-22 23:45:17 · 597 阅读 · 0 评论 -
PyTorch+CUDA+cudnn安装
1、安装CUDA先查看一下电脑显卡对应的CUDA驱动版本上图可以看到CUDA驱动版本为11.1.114,那就下载11.1版本的CUDA好了。(这里应该也不是必须对应,只是说有可能下载低版本的会有问题,没试过,下载对应版本的先跑起来再说)下载地址:https://developer.nvidia.com/cuda-toolkit-archive下载好之后点击文件安装,会先解压到一个临时文件夹内,默认就行然后出现安装步骤我这里就选这四个选项进行安装(没有问...原创 2021-08-22 17:58:29 · 2098 阅读 · 0 评论 -
PyTorch是个啥玩意儿?
1、pytorch卷积2、enumerate()函数,遍历数据3、交叉熵原创 2021-08-21 18:36:33 · 147 阅读 · 0 评论 -
pyinstaller打包的文件运行失败:numpy.core.multiarray failed to import
这个鬼问题,烦死人关键是不打包的文件是可以正常执行的网上的大多数方案主要是说numpy版本问题,我试了好多版本,not work。。待解决原创 2021-02-27 10:24:59 · 363 阅读 · 4 评论 -
PyTorch:将模型转换为torch.jit.ScriptModule
参见上一篇C++调用PyTorch模型import torchimport torchvision# An instance of your model.model = torchvision.models.resnet18()# An example input you would normally provide to your model's forward() method.example = torch.rand(1, 3, 224, 224)# Use torch.ji.原创 2021-02-26 19:54:51 · 2647 阅读 · 4 评论 -
PyTorch:Faster-RCNN
记录一下一个简单的调用Faster-RCNN模型测试的代码coco_names.pynames = {'0': 'background', '1': 'person', '2': 'bicycle', '3': 'car', '4': 'motorcycle', '5': 'airplane', '6': 'bus', '7': 'train', '8': 'truck', '9': 'boat', '10': 'traffic light', '11': 'fire hydrant', '13'原创 2021-02-26 19:48:52 · 259 阅读 · 0 评论 -
C++调用PyTorch模型:LibTorch
转载的文章,挺不错,学习一下!LibTorch学习笔记(一)前天由于某些原因需要利用C++调用PyTorch,于是接触到了LibTorch,配了两天最终有了一定的效果,于是记录一下。环境PyTorch1.6.0cuda10.2opencv4.4.0VS2017具体过程下载LibTorch去PyTorch官网下载LibTorch包,选择对应的版本,这里我选择Stable(1.6.0),Windows,LibTorch,C++/JAVA,10.2,然后我选择release版.转载 2021-02-25 15:42:03 · 3471 阅读 · 6 评论 -
PyTorch:加载自己的数据集
pytorch 加载自己的数据集pytorch 加载自己的数据集,需要写一个继承自torch.utils.data中Dataset类,并修改其中的__init__方法、__getitem__方法、__len__方法。默认加载的都是图片,__init__的目的是得到一个包含数据和标签的list,每个元素能找到图片位置和其对应标签。然后用__getitem__方法得到每个元素的图像像素矩阵和标签,返回img和label。以加载一个图像放在某个文件夹下,并在当前目录下生成了一个.txt的文件,大致如下tr转载 2021-02-22 20:05:40 · 1006 阅读 · 1 评论 -
PyTorch:保存/加载训练好的模型测试
保存torch.save(model.state_dict(), './cnn.pth')加载model = VGG16() #加载模型前要创建一个模型的实例对象model.load_state_dict(torch.load("./cnn.pth"))例子import torchimport torch.nn as nnfrom torch import optimfrom torch.autograd import Variablefrom torch.utils.data原创 2021-02-21 23:29:41 · 3260 阅读 · 0 评论 -
PyTorch:VGG16简单入门版
import torchimport torch.nn as nnfrom torch import optimfrom torch.autograd import Variablefrom torch.utils.data import DataLoaderfrom torchvision import transformsfrom torchvision import datasetsfrom tqdm import tqdm'''定义超参数'''batch_size = 2...原创 2021-02-15 01:01:40 · 699 阅读 · 0 评论 -
PyTorch:VGG16网络模型
深度学习界的小学生一枚从头开始,先了解一下VGG16,哈哈哈VGG16的结构图随便百度一下就好import torch.nn as nnfrom torchsummary import summaryimport torchclass VGG16(nn.Module): def __init__(self): super(VGG16, self).__init__() self.maxpool1 = nn.Sequential(原创 2021-02-08 14:03:32 · 493 阅读 · 0 评论