草庐IT

Pytorch-CUDA

全部标签

【简单作业向】【Pytorch】猫狗分类

【作业向】根据给定的猫狗分类数据集,对比单层CNN模型、从头训练CNN模型(mobileNet)、微调预训练CNN模型(mobileNet)的差异。生成的模型的正向传播图(相关方法见我)。使用PyTorch实现。本文代码(数据集在同目录下):我的Github文章目录关于数据集建立Dataset对象模型1:单层卷积+单层池化+全连接定义训练和评估函数模型2:从头训练(MobileNet)模型3:预训练模型+微调(MobileNet)保存模型前向传播可视化测试集评估模型效果关于数据集数据集结构很简单,训练集和测试集分两个目录,分别对应cat和dog两个文件夹。只需要使用torchvision.da

(超详细)win11换国内源并下载安装pytorch

前言:虽然不是第一次装pytorch,但是这次遇到的问题挺多,不过幸好最后都解决了。目录1.下载Anaconda2.换源2.1生成.condarc文件2.2修改该文件内容 3.去确认下载版本,事半功倍。3.1进入清华大学镜像网站3.2进入之后找到anaconda并依次打开cloud/pytorch/win-64 3.3往下拉,找到自己想要下载的组合3.3.1补充说明确定自己电脑的cuda版本 4.在第三步确定好三个的版本后打开Anacondaprompt,之后所有命令都在这里面4.1创建新的虚拟环境,复制粘贴下面命令。 4.2查看是否创建成功4.3激活刚刚创建的pytorch环境5.进入pyt

Anaconda+PyCharm+PyTorch+Gym深度强化学习环境搭建 送新手直接送进炼丹炉

Part1环境搭建需要下载的软件和包:AnacondaPycharmPythonPyTorchgympygame一、Anaconda下载与安装直接从官网下载:https://www.anaconda.comhttps://www.anaconda.com/点击Download下载即可。下载好后,打开安装包,自己选一个安装路径,默认路径也行,放其他盘也行,我安装在D盘下的Anaconda文件夹下。安装向导一路下一步即可。二、Python和Pycharm下载与安装Python和PyCharm安装过程就不PO了,去官网下载安装即可。附上官网地址:PyCharm:thePythonIDEforProf

深度学习系列5——Pytorch 图像分类(ResNet)

1.概述本文主要是参照B站UP主霹雳吧啦Wz的视频学习笔记,参考的相关资料在文末参照栏给出,包括实现代码和文中用的一些图片。整个工程已经上传个人的githubhttps://github.com/lovewinds13/QYQXDeepLearning,下载即可直接测试,数据集文件因为比较大,已经删除了,按照下文教程下载即可。论文下载:DeepResidualLearningforImageRecognition2.ResNetResNet(deepresidualnetwork)在2015年由微软实验室提出,斩获当年ImageNet竞赛中分类任务第一名,目标检测第一名。获得COCO数据集中目

解决Stable Diffusion TensorRT转换模型报错cpu and cuda:0! (when checking argument for argume

记录StableDiffusionwebUITensorRT插件使用过程的报错:RuntimeError:Expectedalltensorstobeonthesamedevice,butfoundatleasttwodevices,cpuandcuda:0!(whencheckingargumentforargumentmat1inmethodwrapper_CUDA_addmm)拷贝下面的代码覆盖extensions\stable-diffusion-webui-tensorrt里的export_onnx.py文件,将模型和相关的张量移动到GPU,即可解决。importosfrommodu

RuntimeError: FlashAttention is only supported on CUDA 11 and above

RuntimeError:FlashAttentionisonlysupportedonCUDA11andabove问题描述解决方案问题描述RuntimeError:FlashAttentionisonlysupportedonCUDA11andabove解决方案此错误的原因可能是nvcc的CUDA版本(通过键入“nvcc-V”获得,可能类似问题在这里解决:https://stackoverflow.com/questions/40517083/multiple-cuda-versions-on-machine-nvcc-v-confusion我解决了这个问题通过:首先:vim~/.bashr

【新手】复现NeRCo代码中出现的torch.cuda.OutOfMemoryError: CUDA out of memory. 问题解决办法。

代码原文地址:NeRCo问题描述复现CVPR2023中(NeRCo)代码中遇到的问题:torch.cuda.OutOfMemoryError:CUDAoutofmemory.Triedtoallocate26.16GiB(GPU0;14.58GiBtotalcapacity;9.41GiBalreadyallocated;1.32GiBfree;12.25GiBreservedintotalbyPyTorch)Ifreservedmemoryis>>allocatedmemorytrysettingmax_split_size_mbtoavoidfragmentation.Seedocumen

pytorch神经网络对Excel数据集进行处理(读取,转为tensor格式,归一化),并且以鸢尾花(iris)数据集为例,实现BP神经网络

最近跟导师做的项目是关于BP,LSTN神经网络的,数据集对象是一些Excel表格类型的,我使用pytorch进行训练,读取Excel表格数据的时候统一进行一些处理,所以我想把它封装到函数,以后处理其它数据集,直接调用函数实现,这不就方便了吗。我将以鸢尾花数据集作为例子进行展示:我已经编写了2.0版本,方法更加集成化,建议使用2.0版本:2.0可以看到鸢尾花数据集有四个特征,分别是0,1,2,3,label是鸢尾花种类,共三种,分别以0,1,2表示。首先第一部分是读取Excel数据(需要主要的是标签需要在最后一列,函数默认最后一列为标签,前边的为特征值):defopen_excel(filena

ImportError: libtorch_cuda_cu.so: cannot open shared object file: No such file or directory

原因cuda版本选的不对解决python-c'importtorch;print(torch.__version__);print(torch.version.cuda)'查看cuda版本和torch版本我的输出如下:1.9.0+cu10210.2用pip安装时按照上面的选就行refmmcv-fullinstalldoc

【5】深度学习之Pytorch——如何使用张量处理文本数据集(语料库数据集)

在计算机领域,不断崛起的两个领域,一个是CV一个是NLP,下面我们可以探索一下深度学习在NLP的应用和特点。深度学习在自然语言处理(NLP)领域有广泛的应用。以下是一些主要的应用和特点:语音识别:深度学习模型可以通过语音数据训练,学习如何将语音转换为文本。文本分类:深度学习模型可以根据文本内容将文本分为不同的类别。例如,情感分析、主题分类等。机器翻译:深度学习模型可以将一种语言翻译成另一种语言。神经机器翻译是一种基于深度学习的翻译方法。语言生成:深度学习模型可以生成自然语言文本。例如,文本摘要、对话系统等。命名实体识别:深度学习模型可以识别文本中的命名实体,例如人名、地名、组织名等。语言模型: