草庐IT

pytorch-lightning

全部标签

Ubuntu18.04下安装pytorch步骤&多cuda版本共存(啰嗦版)

前言1.知识补充【机器学习】显卡、GPU、NVIDIA、CUDA、cuDNN(搬运:要点如下,详细可看链接)加*非重要内容,视情况执行。显卡:即显示卡,全称显示接口卡,是计算机最基本配置、最重要的配件之一(就像联网需网卡,数据显示在屏幕需显卡)。显卡是由GPU、显存等等组成的。GPU:图形处理器,一般焊接在显卡上的。GPU是显卡上的一块芯片,就像CPU是主板上的一块芯片。GPU功能强大,只用于图形处理太浪费,NVIDIA公司提出CUDA的概念,通用并行计算架构,是一种运算平台,更加方便利用GPU强大的计算能力(并非所有GPU支持CUDA)。CUDA:通用并行计算架构,该架构使GPU能够解决复杂

【深度学习】使用PyTorch实现图像分类(简单Net/VGGNet/AlexNet+官方/自己数据集+plt/混淆矩阵展示统计结果+用训练好的模型进行分类验证)

文章目录使用PyTorch实现图像分类1.定义模型1.1一个小的神经网络1.2AlxeNet网络结构1.3VGG16网络结构2.加载数据集3.定义训练参数4.训练5.显示Loss和Acc5.1使用plot5.2使用混淆矩阵6.验证训练的模型7.问题与解决7.1图像尺寸问题7.2将图像数据划分为训练集、测试集、验证集使用PyTorch实现图像分类本文将介绍如何使用PyTorch实现利用神经网络在图像数据集上进行训练和如何利用训练好的模型对图像进行分类创建文件夹,用于保存训练好的网络importosifnotos.path.exists("./save_model_rs_dataset"):os.

以FGSM算法为例的对抗训练的实现(基于Pytorch)

如果可以,请点个赞,这是我写博客的动力,谢谢各位观众1.前言深度学习虽然发展迅速,但是由于其线性的特性,受到了对抗样本的影响,很容易造成系统功能的失效。以图像分类为例子,对抗样本很容易使得在测试集上精度很高的模型在对抗样本上的识别精度很低。对抗样本指的是在合法数据上添加了特定的小的扰动,人眼不能明显分辨但是会影响深度学习模型的输出的样本。常见的防御方法有对抗训练AdversarialTraining。最近我在尝试复现对抗训练,找了一下发现有一些基于tensorflow的对抗训练的代码,但是没怎么看见pytorch的代码,所以我在这里做一个记录。2.参考文献:1书籍《AI安全之对抗样本入门》2论

Linux 的 anaconda 虚拟环境下安装指定的 cuda、cudnn、pytorch

感悟:首先,anaconda的虚拟环境真香!开辟一个新的虚拟环境,很多环境、版本不兼容的问题都不复存在,尤其对复现别人代码的同学很有用。条件:只要安装的版本不超过自己机器的硬件条件,那么就可以安装。步骤:1.确定安装的cuda版本。在虚拟环境中,先用condasearchcudatoolkit--info命令查看源内所有的cuda版本,以及下载地址。下图示例中展红框标出了支持的cuda版本、对应的url地址及一些约束条件。2.下载并安装cuda。找到你想要的且满足自己机器条件的cuda版本,复制url对应的下载链接,cd到你想要的下载目录,用如下代码下载:wget复制的url链接执行如下命令安

PyTorch深度学习实战 | 预测工资——线性回归

通过员工工作年限与工资的对应关系表,找出二者之间的关系,并预测在指定的年限时,工资会有多少。01、训练流程1●场景说明通过员工工作年限与工资的对应关系表,找出二者之间的关系,并预测在指定的年限时,工资会有多少。可以看出,这是一个用工作年限预测工资的简单线性回归问题。

PyTorch深度学习实战 | 预测工资——线性回归

通过员工工作年限与工资的对应关系表,找出二者之间的关系,并预测在指定的年限时,工资会有多少。01、训练流程1●场景说明通过员工工作年限与工资的对应关系表,找出二者之间的关系,并预测在指定的年限时,工资会有多少。可以看出,这是一个用工作年限预测工资的简单线性回归问题。

PyTorch 深度学习实战 | 基于生成式对抗网络生成动漫人物

生成式对抗网络(GenerativeAdversarialNetwork,GAN)是近些年计算机视觉领域非常常见的一类方法,其强大的从已有数据集中生成新数据的能力令人惊叹,甚至连人眼都无法进行分辨。本文将会介绍基于最原始的DCGAN的动漫人物生成任务,通过定义生成器和判别器,并让这两个网络在参数优化过程中不断“打架”,最终得到较好的生成结果。01、生成动漫人物任务概述日本动漫中会出现很多的卡通人物,这些卡通人物都是漫画家花费大量的时间设计绘制出来的,那么,假设已经有了一个卡通人物的集合,那么深度学习技术可否帮助漫画家们根据已有的动漫人物形象,设计出新的动漫人物形象呢?本文使用的数据集包含已经裁

PyTorch 深度学习实战 | 基于生成式对抗网络生成动漫人物

生成式对抗网络(GenerativeAdversarialNetwork,GAN)是近些年计算机视觉领域非常常见的一类方法,其强大的从已有数据集中生成新数据的能力令人惊叹,甚至连人眼都无法进行分辨。本文将会介绍基于最原始的DCGAN的动漫人物生成任务,通过定义生成器和判别器,并让这两个网络在参数优化过程中不断“打架”,最终得到较好的生成结果。01、生成动漫人物任务概述日本动漫中会出现很多的卡通人物,这些卡通人物都是漫画家花费大量的时间设计绘制出来的,那么,假设已经有了一个卡通人物的集合,那么深度学习技术可否帮助漫画家们根据已有的动漫人物形象,设计出新的动漫人物形象呢?本文使用的数据集包含已经裁

在conda虚拟环境中配置cuda+cudnn+pytorch深度学习环境(新手必看!简单可行!)

本人最近接触深度学习,想在服务器上配置深度学习的环境,看了很多资料后总结出来了对于新手比较友好的配置流程,创建了一个关于深度学习环境配置的专栏,包括从anaconda到cuda到pytorch的一系列操作,专栏中的另外两篇文章如下,如果有不对的地方欢迎大家批评指正!Anaconda保姆级安装配置教程(新手必看)如果你还不是很清楚CUDA,CUDATookit,cuDNN,Pytorch分别在深度学习环境中的定位与关系,推荐看一下我这篇文章用人话讲解深度学习中CUDA,cudatookit,cudnn和pytorch的关系另外,如果你还不是特别理解虚拟环境的作用,建议看一下这位大佬的视频安装不算

在conda虚拟环境中配置cuda+cudnn+pytorch深度学习环境(新手必看!简单可行!)

本人最近接触深度学习,想在服务器上配置深度学习的环境,看了很多资料后总结出来了对于新手比较友好的配置流程,创建了一个关于深度学习环境配置的专栏,包括从anaconda到cuda到pytorch的一系列操作,专栏中的另外两篇文章如下,如果有不对的地方欢迎大家批评指正!Anaconda保姆级安装配置教程(新手必看)如果你还不是很清楚CUDA,CUDATookit,cuDNN,Pytorch分别在深度学习环境中的定位与关系,推荐看一下我这篇文章用人话讲解深度学习中CUDA,cudatookit,cudnn和pytorch的关系另外,如果你还不是特别理解虚拟环境的作用,建议看一下这位大佬的视频安装不算