随笔分类 - Pytorch
摘要：setuptools 版本过高，需要降级，降为59.5.0版本比较合适： pip install setuptools==59.5.0 参考：解决AttributeError: module ‘setuptools._distutils‘ has no attribute ‘version‘_set
阅读全文
摘要：Pytorch自动混合精度(AMP)介绍与使用
阅读全文
摘要：【PyTorch踩坑】一个排查了一下午的坑 - 知乎 (zhihu.com)
阅读全文
摘要：import torch box = torch.tensor([ # 边界框的坐标，(x1, y1, x2, y2). box'shape: (3, 4) [0.1, 0.2, 0.5, 0.3], [0.6, 0.6, 0.9, 0.9], [0.1, 0.1, 0.2, 0.2] ]) whw
阅读全文
摘要：本文摘自：知乎 用Pytorch中查看GPU信息 1. 返回当前设备索引 torch.cuda.current_device() 2. 返回GPU的数量 torch.cuda.device_count() 3. 返回gpu名字，设备索引默认从0开始 torch.cuda.get_device_nam
阅读全文
摘要：PyTorch Python中的注册器模块 MMCV 核心组件分析(五): Registry Detectron2和fvcore中的Registry机制详解
阅读全文
摘要：参考：端口冲突(Address already in use)解决方法
阅读全文
摘要：yacs的使用小记，merge_from_file()
阅读全文
摘要：如何使用 PyTorch Hook
阅读全文
摘要：mmcv的安装会有很多cuda、pytorch版本适配的问题，如果版本不适配，会报各种错误，比如ModuleNotFoundError: No module named 'mmcv._ext' 。 mmdetection官方建议用mim进行安装，这是官方的整个安装教程：https://mmdetec
阅读全文
摘要：pytorch教程之nn.Module类详解——使用Module类来自定义模型
阅读全文
摘要：pytorch函数（1）：torch.gather()的正确理解方法
阅读全文
摘要：参考：在pytorch模型中获取权重和偏差并将其复制到另一个模型中的类似层的正确方法是什么？
阅读全文
摘要：pytorch张量维度操作（拼接、维度扩展、压缩、转置、重复等）
阅读全文
摘要：本文的内容摘自以下资料： PyTorch分布式训练简明教程 （知乎，推荐） PyTorch分布式训练基础--DDP使用 （知乎，推荐） Pytorch DistributedDataParallel简明使用指南 （知乎） pytorch多机多卡分布式训练（知乎） 详解PyTorch FSDP数据并行
阅读全文
摘要：torch.backends.cudnn.benchmark (推荐，讲解的很详细) cuDNN 是英伟达专门为深度神经网络所开发出来的 GPU 加速库，针对卷积、池化等等常见操作做了非常多的底层优化，比一般的 GPU 程序要快很多。大多数主流深度学习框架都支持 cuDNN，PyTorch 自然也不
阅读全文
摘要：pytorch之多GPU使用——#CUDA_VISIBLE_DEVICES使用 #torch.nn.DataParallel() #报错解决
阅读全文
摘要：本文的参考资料： Pytorch——多卡GUP训练原理(torch.nn.DataParallel) （博客园） [源码解析] PyTorch 分布式(2) DataParallel(上) Pytorch的nn.DataParallel（知乎） DataParallel的流程 DataParalle
阅读全文
摘要：nn.RNN和nn.RNNCell的使用 nn.LSTM和nn.LSTMCell的使用
阅读全文
