pytorch
-
Linux 的 anaconda 虚拟环境下安装指定的 cuda、cudnn、pytorch
感悟：首先，anaconda 的虚拟环境真香！开辟一个新的虚拟环境，很多环境、版本不兼容的问题都不复存在，尤其对复现别人代码的同学很有用。 条件：只要安装的版本不超过自己机器的硬件…
-
PyTorch学习笔记：F.normalize——数组归一化运算
PyTorch学习笔记：F.normalize——数组归一化运算 torch.nn.functional.normalize(input, p=2.0, dim=1, eps=1e…
-
对 ChatGLM-6B 做 LoRA Fine-tuning
对 ChatGLM-6B 做 LoRA Fine-tuning 搭建依赖环境 加载模型和 Tokenizer 分析模型结构 配置 LoRA 构建数据集 定义常量 测试 Tokeni…
-
报错解决：RuntimeError: Error compiling objects for extension和nvcc fatal: Unsupported gpu architecture
报错解决：RuntimeError: Error compiling objects for extension和nvcc fatal: Unsupported gpu archi…
-
分类任务使用Pytorch实现Grad-CAM绘制热力图
对于深度学习网络，在我们指定数据集类别的情况下，Grad-CAM能够绘制出相应的热力图，让我们能够非常直观的看出网络关注的主要区域与特征是什么。本文主要记录在绘制热力图过程中，自己…
-
轴承故障诊断经典模型pytorch复现(一)——WDCNN
论文地址：《A New Deep Learning Model for Fault Diagnosis with Good Anti-Noise and Domain Adapta…
-
YOLOV7改进-添加EIoU,SIoU,AlphaIoU,FocalEIoU,Wise-IoU
在YoloV7中添加EIoU,SIoU,AlphaIoU,FocalEIoU,Wise-IoU. yolov7中box_iou其默认用的是CIoU，其中代码还带有GIoU，DIoU…
-
【NLP相关】PyTorch多GPU并行训练（DataParallel和DistributedDataParallel介绍、单机多卡和多机多卡案例展示）
❤️觉得内容不错的话，欢迎点赞收藏加关注😊😊😊，后续会继续输入更多优质内容❤️ 👉有问题欢迎大家加关注私戳或者评论（…
-
一行代码解决PyTorch训练模型时突然出现的For debugging consider passing CUDA
一、问题描述 今天在调试模型的代码，然后代码一直运行得好好地，就突然出现了一下的错误： RuntimeError: CUDA error: invalid devi…
-
Win11基于WSL2安装CUDA、cuDNN和TensorRT（2023-03-01）
文章目录 2023-03-06 更新 2023-03-05 更新 前言 TensorRT介绍 环境配置 CUDA Driver 检查安装 安装CUDA 安装nvcc 安装cuDNN…
-
YOLOv5+DeepSORT多目标跟踪与计数算法
YOLOv5+DeepSORT多目标跟踪与计数算法 1、协方差数学求解方式 2、马氏距离求解方式： 3、匈牙利算法–用来解决分配、指派问题 4、卡尔曼滤波器 5、卡尔曼…
-
RuntimeError: CUDA error: CUBLAS
今天跑一个项目时遇到了如下问题： RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cub…
-
关于安装 PyTorch-Lightning 的一些问题（GPU版）
官网地址： PyTorch PyTorch-Lightning 安装PyTorch-Lightning 1、不能直接使用pip install pytorch-lightning …
-
使用PyTorch构建GAN生成对抗网络源码（详细步骤讲解+注释版）01 手写字体识别
文章目录 1 生成对抗网络基本概念 2 生成对抗网络建模 2.1 建立MnistDataset类 2.2 建立鉴别器 2.3 测试鉴别器 2.4 Mnist生成器制作 3 模型的训…
-
自注意力(Self-Attention)与Multi-Head Attention机制详解
自注意力机制属于注意力机制之一。与传统的注意力机制作用相同，自注意力机制可以更多地关注到输入中的关键信息。self-attention可以看成是multi-head atten…
