Pytorch 的实际应用 学习笔记
一. 模型的下载 weights为false时则为没有提前经过训练的模型，为true时则经过了提前训练 vgg16_false = torchvision.models.vgg16(weights=False)vgg16_true = torchvision.models.vgg16(weights=True) 打印 二. 模型的修改 （1）添加操作 分为两种，一种是在classifier的外部添加，一...
拦截pytorch算子,dump输入输出
拦截pytorch算子,dump输入输出 一.代码二.输出 希望dump出pytorch每个算子的输入输出,但pytorch普通的hook机制只能拦截module.以下提供一种方法可以拦截torch.add,torch.Tensor.add这类算子.原理是通过模板替换,劫持torch和torch.Tensor中的算子.遍历next_functions调用register_hook拦截backward. 一...
卸载并升级pytorch安装torcheval
Requires Python >= 3.8 and PyTorch >= 1.11conda list torch 查看包 pip uninstall torch 及各个包，nvidia-smi查看cuda11.4，安装对应11.3去官网找cuda11.3和torch1.11.0并粘贴命令，再安装torcheval pip install torch0.12.0+cu113 torchaudio==0....
GPU深度学习环境搭建：Win10+CUDA 11.7+Pytorch1.13.1+Anaconda3+python3.10.9
1. 查看显卡驱动及对应cuda版本关系 1.1 显卡驱动和cuda版本信息查看方法 在命令行中输入【nvidia-smi】可以当前显卡驱动版本和cuda版本。 根据显示，显卡驱动版本为：Driver Version: 516.59，CUDA 的版本为：CUDA Version 11.7。 此处我们可以根据下面的表1 显卡驱动和cuda版本对应关系得知，Driver Version: 527.47 >...
【PyTorch与深度学习】5、深入剖析PyTorch DataLoader源码
课程地址 最近做实验发现自己还是基础框架上掌握得不好，于是开始重学一遍PyTorch框架，这个是课程笔记，此节课很详细，笔记记的比较粗 1. DataLoader 1.1 DataLoader类实现 1.1.1 构造函数__init__实现 构造函数有如下参数： dataset：传入自己定义好的数据集类Datasetbatch_size：默认值为1，它代表着每批次训练的样本的个数shuffle：布尔类型，...
【图像超分】论文复现：Pytorch实现RDN！保姆级复现教程！实现与原论文基本一致的PSNR和SSIM！代码注释详尽！易读易复用！可用于训练自己的数据集！附完整代码和各放大倍数下的最优模型权重文件！
第一次来请先看这篇文章：【超分辨率（Super-Resolution）】关于【超分辨率重建】专栏的相关说明，包含专栏简介、专栏亮点、适配人群、相关说明、阅读顺序、超分理解、实现流程、研究方向、论文代码数据集汇总等） 项目代码和最优性能的权重文件下载见文末链接！！！可以复现论文、训练自己的数据集，详细代码使用说明！！！同时包含最优性能的模型权重文件（x2、x3、x4），可以直接用来超分自己的图像！！！ 本文...
torchvision指定版本whl安装（Ubuntu20环境）
pytorch教程需要torchvision下载数据集，使用pip安装指定版本，首先使用conda list torch查看自己安装torch版本，我的pytorch版本1.9.0对应cuda版本11.1 在以下网址查找对应torchvision版本，https://pytorch.org/get-started/previous-versions/，pytorch0.10.0 再去一下网站下载对应whl...
基于pytorch hook机制,生成一次前后向的timeline
基于pytorch hook机制,生成一次前后向的timeline 一.效果图二.代码三.运行以上代码,生成timeline.json,打开https://ui.perfetto.dev/,导入该文件,可看到以上效果 本文通过pytorch hook机制,拦截所有子module的forward和backward,统计执行时间及内存使用情况 一.效果图 二.代码 import osos.environ["...
Pytorch的下载安装
本文为自己整理的Pytorch下载相关的内容笔记，以便日后查阅 一. 基本命令 1.查看conda版本 conda --version 2.创建conda新环境 conda create –n 名称 python=版本 3.查看已经创建的conda环境 conda info --envs 4.进入虚拟环境 conda activate 环境名称 为了避免以前的环境python版本与现在不二. 下载Pytor...
Ubuntu20从0开始选择合适版本手动安装cuda，torch-geometric，jax
一个全新的ubuntu20台式机，在Additional Drivers安装nvidia-470-server（一开始安装450，cunda版本只能到11.0，torch有些库用不了，可以直接切换点击Apply Changes重启就行） nvidia-smi查看CUDA Version可到11.4，手动安装11.2，以下地址下载并安装，安装完成后加入环境变量，用nvcc -V测试是否安装成功 https:...