草庐IT

pytorch-cuda

全部标签

python - 如何将 MNIST 图像加载到 Pytorch DataLoader 中?

用于数据加载和处理的pytorch教程非常具体到一个示例,有人可以帮助我了解更通用的简单图像加载函数应该是什么样子吗?教程:http://pytorch.org/tutorials/beginner/data_loading_tutorial.html我的数据:我在以下文件夹结构中有jpg格式的MINST数据集。(我知道我可以只使用数据集类,但这纯粹是为了了解如何在没有csv或复杂功能的情况下将简单图像加载到pytorch中)。文件夹名称是标签,图像是28x28png的灰度图像,不需要转换。datatrain03.png5.png13.png23.png...13.png10.png1

PyTorch 进行多步时间序列预测详细教程

一、前言Encoder-decoder模型提供了最先进的结果,可以对语言翻译等NLP任务进行排序。多步时间序列预测也可以视为seq2seq任务,可以使用编码器-解码器模型。本文提供了一个Encoder-decoder模型来解决Kaggle的时间序列预测任务以及获得前10%结果所涉及的步骤。模型实现灵感来自Pytorchseq2seq翻译教程,时间序列预测思路主要来自Kaggle类似比赛的获奖方案。二、数据使用的数据集来自过去的Kaggle竞赛——StoreItem需求预测挑战,给出过去5年(从2013年到2017年)来自10家不同商店的50件商品的销售数据,预测未来3个月内每件商品的销量(01

python - 让 TensorFlow 使用由自定义 CUDA 例程即时生成的训练数据

假设我们生成自己的训练数据(例如,通过从一些扩散过程中采样并计算一些感兴趣的数量)并且我们有自己的CUDA例程,称为generate_data,它在GPU内存中为给定的一组输入生成标签.因此,我们处于一个特殊的环境中,我们可以以“在线”方式生成任意数量的训练数据批处理(在每次批处理迭代中,我们调用generate_data例程来生成新批处理并丢弃旧批处理).既然数据是在GPU上生成的,有没有办法让TensorFlow(PythonAPI)在训练过程中直接使用它?(例如填充占位符)这样,这样的管道会很有效。我的理解是,目前您需要在这样的设置中将数据从GPU复制到CPU,然后让Tensor

ubuntu18.04 cuda卸载及安装

1.若电脑上已经安装了其他版本的cuda及显卡驱动,需要完全卸载并删除相关文件,否则会导致安装不成功,执行如下:1.1卸载cuda,步骤如下:cd/usr/local/cuda-xx.x/bin/(进入你的cuda文件夹下)sudo./cuda-uninstallersudorm-rf/usr/local/cuda-xx.x(删除cuda文件夹)1.2卸载驱动,步骤如下:sudoapt-getremove--autoremovenvidia-cuda-toolkitsudoapt-getpurgenvidia-cuda-toolkit若上面的命令无效,则执行:sudonvidia-uninst

ubuntu18.04 cuda卸载及安装

1.若电脑上已经安装了其他版本的cuda及显卡驱动,需要完全卸载并删除相关文件,否则会导致安装不成功,执行如下:1.1卸载cuda,步骤如下:cd/usr/local/cuda-xx.x/bin/(进入你的cuda文件夹下)sudo./cuda-uninstallersudorm-rf/usr/local/cuda-xx.x(删除cuda文件夹)1.2卸载驱动,步骤如下:sudoapt-getremove--autoremovenvidia-cuda-toolkitsudoapt-getpurgenvidia-cuda-toolkit若上面的命令无效,则执行:sudonvidia-uninst

python - Pytorch:将 FloatTensor 转换为 DoubleTensor

我有2个numpy数组,我将它们转换为张量以使用TensorDataset对象。importtorch.utils.dataasdata_utilsX=np.zeros((100,30))Y=np.zeros((100,30))train=data_utils.TensorDataset(torch.from_numpy(X).double(),torch.from_numpy(Y))train_loader=data_utils.DataLoader(train,batch_size=50,shuffle=True)当我这样做时:forbatch_idx,(data,target)i

面部表情识别2:Pytorch实现表情识别(含表情识别数据集和训练代码)

面部表情识别2:Pytorch实现表情识别(含表情识别数据集和训练代码)目录面部表情识别2:Pytorch实现表情识别(含表情识别数据集和训练代码)1.面部表情识别方法2.面部表情识别数据集 (1)表情识别数据集说明 (2)自定义数据集3.人脸检测模型4.面部表情识别分类模型训练(1)项目安装(2)准备数据(3)面部表情识别分类模型训练(Pytorch)(4)可视化训练过程(5)面部表情识别效果(6)一些优化建议(7)一些运行错误处理方法5.项目源码下载(Python版)6.项目源码下载(Android版)这是项目《面部表情识别》系列之《Pytorch实现表情识别(含表情识别数据集和训练代码)

面部表情识别2:Pytorch实现表情识别(含表情识别数据集和训练代码)

面部表情识别2:Pytorch实现表情识别(含表情识别数据集和训练代码)目录面部表情识别2:Pytorch实现表情识别(含表情识别数据集和训练代码)1.面部表情识别方法2.面部表情识别数据集 (1)表情识别数据集说明 (2)自定义数据集3.人脸检测模型4.面部表情识别分类模型训练(1)项目安装(2)准备数据(3)面部表情识别分类模型训练(Pytorch)(4)可视化训练过程(5)面部表情识别效果(6)一些优化建议(7)一些运行错误处理方法5.项目源码下载(Python版)6.项目源码下载(Android版)这是项目《面部表情识别》系列之《Pytorch实现表情识别(含表情识别数据集和训练代码)

PyTorch 之 基于经典网络架构训练图像分类模型

文章目录一、模块简单介绍1.数据预处理部分2.网络模块设置3.网络模型保存与测试二、数据读取与预处理操作1.制作数据源2.读取标签对应的实际名字3.展示数据三、模型构建与实现1.加载models中提供的模型,并且直接用训练的好权重当做初始化参数2.参考pytorch官网例子3.设置哪些层需要训练4.优化器设置5.训练模块6.测试模型效果本文参加新星计划人工智能(Pytorch)赛道:https://bbs.csdn.net/topics/613989052一、模块简单介绍我们可以进入pytorch的官方网站,对模型的基本架构和训练好的参数进行直接调用,具体链接如下https://pytorch

PyTorch 之 基于经典网络架构训练图像分类模型

文章目录一、模块简单介绍1.数据预处理部分2.网络模块设置3.网络模型保存与测试二、数据读取与预处理操作1.制作数据源2.读取标签对应的实际名字3.展示数据三、模型构建与实现1.加载models中提供的模型,并且直接用训练的好权重当做初始化参数2.参考pytorch官网例子3.设置哪些层需要训练4.优化器设置5.训练模块6.测试模型效果本文参加新星计划人工智能(Pytorch)赛道:https://bbs.csdn.net/topics/613989052一、模块简单介绍我们可以进入pytorch的官方网站,对模型的基本架构和训练好的参数进行直接调用,具体链接如下https://pytorch