草庐IT

tiny-cuda-nn

全部标签

基于OpenCV+CUDA实时视频抠绿、背景合成以及抠绿算法小结

一、关于抠绿百度百科上描述抠绿“抠绿是指在摄影或摄像时,以绿色为背景进行拍摄,在后期制作时使用特技机的“色键”将绿色背景抠去,改换其他更理想的背景的技术。”绿幕的使用已经非常普遍,大到好莱坞大片,小到自媒体的节目,一些商业娱乐场景,几乎都用使用。但是很多非专业场景中经常会出现一些错误,导致最后的结果并不理想。我们这里会对于影响抠绿的条件做一些简要描述,然后看几种背景抠绿的简单算法。二、抠绿算法1、基于HSV颜色表这种方式opencv、GpuMat版本参考代码

基于OpenCV+CUDA实时视频抠绿、背景合成以及抠绿算法小结

一、关于抠绿百度百科上描述抠绿“抠绿是指在摄影或摄像时,以绿色为背景进行拍摄,在后期制作时使用特技机的“色键”将绿色背景抠去,改换其他更理想的背景的技术。”绿幕的使用已经非常普遍,大到好莱坞大片,小到自媒体的节目,一些商业娱乐场景,几乎都用使用。但是很多非专业场景中经常会出现一些错误,导致最后的结果并不理想。我们这里会对于影响抠绿的条件做一些简要描述,然后看几种背景抠绿的简单算法。二、抠绿算法1、基于HSV颜色表这种方式opencv、GpuMat版本参考代码

【nn.LSTM详解】

参数详解nn.LSTM是pytorch中的模块函数,调用如下:torch.nn.lstm(input_size,hidden_size,num_layers,bias,batch_first,dropout,bidirectional)详细介绍一下参数:input_size:表示的是输入的矩阵特征数,或者说是输入的维度;hidden_size:隐藏层的大小(即隐藏层节点数量),输出向量的维度等于隐藏节点数;num_layers:lstm隐层的层数,默认为1;bias:隐层状态是否带bias,默认为true;batch_first:True或者False,如果是True,则input为(batc

【nn.LSTM详解】

参数详解nn.LSTM是pytorch中的模块函数,调用如下:torch.nn.lstm(input_size,hidden_size,num_layers,bias,batch_first,dropout,bidirectional)详细介绍一下参数:input_size:表示的是输入的矩阵特征数,或者说是输入的维度;hidden_size:隐藏层的大小(即隐藏层节点数量),输出向量的维度等于隐藏节点数;num_layers:lstm隐层的层数,默认为1;bias:隐层状态是否带bias,默认为true;batch_first:True或者False,如果是True,则input为(batc

PyTorch、CUDA Toolkit 及显卡驱动版本对应关系

CUDA驱动及CUDAToolkit最高对应版本如下:如果上述没有你想要的,参考官方文档注:驱动是向下兼容的,其决定了可安装的CUDAToolkit的最高版本。CUDAToolkit版本及可用PyTorch对应关系注:虽有的卡驱动更新至较新版本,且CUDAToolkit及PyTorch也可对应更新至新版本。但可能用不了,建议用稳定版查询驱动版本nvidia-smi此处提供三种方法可供选择。(1)指定CUDAToolkit版本(推荐)根据表一查询到可安装的CUDAToolkit版本,470.141对应最高的CUDAToolkit版本为11.4。上述表格没有,你可以装11.3的驱动,向下兼容向上不

PyTorch、CUDA Toolkit 及显卡驱动版本对应关系

CUDA驱动及CUDAToolkit最高对应版本如下:如果上述没有你想要的,参考官方文档注:驱动是向下兼容的,其决定了可安装的CUDAToolkit的最高版本。CUDAToolkit版本及可用PyTorch对应关系注:虽有的卡驱动更新至较新版本,且CUDAToolkit及PyTorch也可对应更新至新版本。但可能用不了,建议用稳定版查询驱动版本nvidia-smi此处提供三种方法可供选择。(1)指定CUDAToolkit版本(推荐)根据表一查询到可安装的CUDAToolkit版本,470.141对应最高的CUDAToolkit版本为11.4。上述表格没有,你可以装11.3的驱动,向下兼容向上不

GPU版本的pytorch安装(显卡为3060ti,如何选择对应的cuda版本)

文章目录前言一、基本知识二、安装步骤1.首先判断自己有无英伟达的显卡2.安装或者查看自己的显卡驱动3.显卡的算力必须与Cudaruntimeversion相匹配4.根据以上两条来选择合适Cudaruntimeversion5.下载pytorch前言显卡为3060tig6x,操作系统win10一、基本知识要清楚下面的几个常识1.GPU和CPU是采用不同架构设计出来的,简单来说,GPU会比CPU多很多计算单元,用于训练网络时,速度比CPU快很多。2.CUDA是一个计算平台和编程模型,提供了操作GPU的接口。3.网上很多教程说的安装CUDA其实是指CUDAToolkit,是一个工具包4.CUDNN是

解决cuda突然不能用了的方法[完美解决]

一、问题描述之前配置过一个pytoch的虚拟环境,并且调用cuda也可用,但是在做项目的时候,想下载一些工具包,下载时出现了一些问题,上网解决时候告诉我利用condaupdateall更新conda即可,于是我更新了,至此之后我的cuda就不可用了,出现torch.cuda.is_available()返回False的问题 二、问题分析之前我的虚拟环境中cudNN可用,但是cuda在pytorch调用不出来,大概率还是pytorch版本对应的问题,就是之前下载的pytorch版本是1.12.1对应cuda11.3,然后我更新了condaupdateall,这时我更新后的pytorch版本是1.

成功解决:AssertionError: Torch not compiled with CUDA enabled

在运行pycharm项目的时候,出现了以上的报错,主要可以归结于以下两个个方面:1、没有安装GPU版本的pytorch,只是使用清华的镜像地址下载了CPU版本的pytorch2、安装的CUDA和安装的pytorch的版本不相互对应我使用  piplist来查看我在该环境下安装了哪些依赖项,发现自己的torch是CPU版本的, 如何区分的呢?主要通过安装的环境的版本号的后面时候加上了cu,比如说下面这幅图就可以看出这里采用的CUDA版本号为11.3,所以在我的环境里面没有安装GPU版本的torch. 找到问题后,我们就可以进行解决了。如果自己不是CUDA版本的torch,可以使用pinunins

[踩坑记] CUDA环境下bitsandbytes安装报错/异常解决

关键词:模型量化,bitsandbytes,bitsandbytes报错,大模型环境,CUDA环境1.问题描述在部署大模型LLaMA的过程中,需要安装量化工具包bitsandbytes,环境如下:操作系统:Ubuntu18.04GPU:4xA10040GCUDA:11.7cuDNN:8.4bitsandbytes:0.38.0github库在condapython环境下,通过pipinstallbitsandbytes直接安装,或是按源码编译安装,在python-mbitsandbytes检测时均出现异常。详见ISSUES),主要异常内容如下:File'bitsandbytes/cextens