jn10010537
2024年,C#/C++
展开
-
【PyTorch】 多进程队列中传入pytorch处理后的tensor张量
【PyTorch】 多进程队列中传入pytorch处理后的tensor张量1、背景2、说明3、示例1、背景视频监控任务中,涉及到解码、算法、编码推流,经常需要使用多进程。以下是多进程中传输的pytorch的tensor的报错示例:# coding=utf-8# yusimport multiprocessingimport numpy as npimport torchdef algo(que): print("调用algo函数 ... ...") data = que原创 2022-03-19 11:21:55 · 1898 阅读 · 2 评论 -
【PyTorch】安装支持cuda的pytorch-1.10.2
【PyTorch】安装支持cuda的pytorch-1.10.21、背景2、安装支持cuda的pytorch 1.10.23、验证pytorch是否可用cuda4、使用whl轮子加速安装1、背景当下流行的深度学习框架pytorch,官网:https://pytorch.org/get-started/locally/其官网上有详细的安装教程,如下是Stable(1.10.2)2、安装支持cuda的pytorch 1.10.2Pytorch支持cuda的Stable(1.10.2),其pip安装原创 2022-03-16 21:57:41 · 5781 阅读 · 0 评论 -
【PyTorch】numpy数组与pytorch的tensor相互转化
【PyTorch】numpy数组与pytorch的tensor相互转化1、背景2、numpy 转为 pytorch3、pytorch 转为 numpy1、背景numpy是python做数值运算最常用的三方包,很多著名的三方包比如opencv也依赖numpy。本博客以示例的形式说明:numpy数组与pytorch的tensor相互转化2、numpy 转为 pytorch使用: torch.from_numpy(np_data)代码示例:import torchprint(torch.__ve原创 2021-09-14 05:02:46 · 860 阅读 · 0 评论 -
【PyTorch】 tensor.squeeze压缩维度
【PyTorch】 tensor压缩一个维度1、背景2、squeeze方法3、示例1、背景pytorch的tensor维度扩展、压缩是一个常见的操作,tensor维度扩展参见如下博客:【PyTorch】 tensor扩展一个维度https://jn10010537.blog.csdn.net/article/details/120303933本博客tensor压缩一个维度的方法:squeeze,即压缩。关于unsqueeze方法的说明如下:import torchprint(torch._原创 2021-09-17 20:27:43 · 3217 阅读 · 0 评论 -
【PyTorch】取出tensor中的值
【PyTorch】取出tensor中的值1、背景2、示例3、运行1、背景在使用pytorch进行运算时,pytorch中数据经常以tensor形式出现。如果你还不熟悉tensor的操作,可以将tensor的值取出来做分析。操作关键字:tensor.item()2、示例import torchprint(torch.__version__)a = torch.rand(3, 2)print(a)print()print("*"*100)for i in a: print(i)原创 2021-09-14 04:28:24 · 15805 阅读 · 0 评论 -
【PyTorch】 yolov5推理
【PyTorch】 yolov5推理1、背景2、Inference with YOLOv5 and PyTorch Hub3、Inference with detect.py1、背景参考官方代码托管地址https://github.com/ultralytics/yolov5如下图1:在此,非常感谢ultralytics公司的伟大开源精神!2、Inference with YOLOv5 and PyTorch HubInference with YOLOv5 and PyTorch Hub.原创 2021-08-29 13:58:03 · 608 阅读 · 0 评论 -
【PyTorch】 查看张量的数据类型
【PyTorch】 查看张量的数据类型1、背景2、dtype和type()的区别3、示例1、背景我们在测试硬件上推理的时候,经常需要知道什么数据类型下的运行速度。查看张量的数据类型可以通过:torch_tensor.dtype或者torch_tensor.type()2、dtype和type()的区别torch_tensor.dtype返回类似: torch.float64,是torch定义的数据类型,详见示例。torch_tensor.type()返回类似:torch.cuda.Dou原创 2021-11-18 20:45:09 · 3335 阅读 · 0 评论 -
【PyTorch】 转换张量的数据类型
【PyTorch】 转换张量的数据类型1、背景2、独立的函数转换数据类型3、使用type()函数转换数据类型4、使用type_as()函数转换数据类型1、背景Pytorch的tensor数据类型转换方法:使用独立的函数如 int(),float()等进行转换;使用torch.type()函数,直接显示输入需要转换的类型;使用type_as()函数,将该tensor转换为另一个tensor的type;2、独立的函数转换数据类型常用的如下:数据类型转化torch_tensor.dtyp原创 2021-11-18 21:03:13 · 2365 阅读 · 0 评论 -
【PyTorch】 tensor.unsqueeze扩展一个维度
【PyTorch】 给tensor扩展一个维度1、背景2、unsqueeze方法3、示例1、背景在tensor合并的时候,经常需要考虑tensor的维度的对齐,即涉及维度的扩展;pytorch中关于维度扩展的方法:unsqueeze(…)关于unsqueeze方法的说明如下:import torchprint(torch.__version__)help(torch.unsqueeze)运行结果如下:Help on built-in function unsqueeze:unsque原创 2021-09-15 12:27:35 · 1227 阅读 · 0 评论 -
【PyTorch】 torch版本1.9.0+cu102能否使用CUDA 10.1
【torch】 torch版本1.9.0+cu102能否使用CUDA 10.11、背景2、安装torch3、验证GPU4、总结1、背景torch和tensorflow两大主流的深度学习训练框架(tensorflow框架现在似乎开始没落了)。参考torch官网:https://pytorch.org/get-started/locally/参考tensorflow官网:https://tensorflow.google.cn/install/source#gpu就是这么不和谐。比如, tenso原创 2021-08-18 18:22:41 · 4200 阅读 · 2 评论 -
【PyTorch】 查看cuda以及cuda的gpu
【PyTorch】 查看cuda以及cuda的gpu1、cuda是否可用2、cuda的gpu数量3、cuda的gpu名称4、cuda当前的gpu索引1、cuda是否可用接口:torch.cuda.is_available()如下示例,cuda可用:root@3dd53ab5928d:/# python3Python 3.6.9 (default, Jan 26 2021, 15:33:00) [GCC 8.4.0] on linuxType "help", "copyright", "cred原创 2021-10-09 21:30:00 · 2156 阅读 · 0 评论 -
【PyTorch】 tensor.cat张量合并
【PyTorch】 tensor合并1、背景2、cat方法3、示例1、背景pytorch中tensor合并方法是cat,即torch.cat:将2个及以上的tensor合并为一个tensor。打印官方关于torch.cat的说明:import torchprint(torch.__version__)help(torch.cat)运行如下:2、cat方法具体如下:def cat(tensors: Union[Tuple[Tensor, ...], List[Tensor]], di原创 2021-09-21 23:02:48 · 3960 阅读 · 0 评论 -
【LibTorch】 windows10无法定位程序输入点cusparseGetErrorString于动态链接库
【torch】无法定位程序输入点xxx于动态链接库一级目录二级目录三级目录一级目录msvcrt.dll是微软在windows操作系统中提供的c语言运行库执行文件(MicrosoftCRuntimeLibrary),其中提供了printf,malloc,strcpy等C语言库函数的具体运行实现,并且为使用C/C++(Vc)编绎的程序提供了初始化(如获取命令行参数)以及退出等功能。二级目录如果您的系统提示“没有找到MSVCRT.DLL”或者“缺少MSVCRT.DLL”等类似错误信息,请把MSVCRT.D原创 2021-08-21 11:48:16 · 1242 阅读 · 2 评论 -
【LibTorch】Microsoft C++ 异常: c10::NotImplementedError,位于内存位置 0x000000E8A9DAEDC0 处。
【torch】Microsoft C++ 异常: c10::NotImplementedError,位于内存位置 0x000000E8A9DAEDC0 处。1、背景2、解决办法1、背景最近学习libtorch(pytorch的c++版本)。开发环境,参考博客:【torch】 windows10无法定位程序输入点cusparseGetErrorString于动态链接库https://jn10010537.blog.csdn.net/article/details/119809829Visual S原创 2021-08-21 15:16:21 · 3718 阅读 · 2 评论 -
【LibTorch】 检测LibTorch-GPU是否可用
【LibTorch】 判定GPU是否可用1、背景2、代码1、背景最近学习libtorch(pytorch的c++版本)。开发环境介绍:-1.1 windows10;-1.2 Microsoft Visual Studio 2017社区版;-1.3 libTorch: libtorch-win-shared-with-deps-1.9.0+cu102.zip-1.4 cuda: cuda_10.2.89_441.22_win10.exe-1.5 cudnn: cudnn-10.2-window原创 2021-08-21 17:58:26 · 928 阅读 · 0 评论 -
【LibTorch】 windows10下vs2017配置LibTorch
【LibTorch】 windows10下vs2017加载LibTorch0、背景1、vs2017新建项目2、配置:包含目录3、配置:库目录4、配置:附加依赖项5、测试libtorch0、背景libTorch、cuda配置按照官方进行: https://pytorch.org/get-started/locally/本博客下载的是Release版本。本博客涉及的PC系统以及软件包如下:windows10Windows10专业版Visual Studiovs2017社区版原创 2021-11-28 15:56:20 · 276 阅读 · 0 评论