pytorch
-
PyTorch中torch、torchtext、torchvision、torchaudio与Python版本兼容性
torch 和 torchtext 版本适配 torch与torchtext，Python对应关系，来源：https://pypi.org/project/torchtext/ 截…
-
Python的Registry机制及PyTorch中的基础应用（注册、Register、配置、配置文件、cfg、config、conf、.cfg、.config、.conf）
1. 前言 注册机制是一种在编程中常见的设计模式，它允许程序在运行时动态地将函数、类或其他对象注册到某个中心管理器中，以便随后可以使用这些注册的对象。在Python中，注册机制通常…
-
pytorch模型运行到android手机上(仅使用pytorch+AndroidStudio)
近期需要将pytorch模型运行到android手机上实验，在查阅网上博客后，发现大多数流程需要借助多个框架或软件，横跨多个编程语言、IDE。本文参考以下两篇博文，力求用更简洁的流…
-
AMD GPU驱动,ROCM,Pytorch安装教程(A卡6700xt)
我用的操作系统为ubuntu20.04,其他系统应该类似，只是命令稍有不同。 安装AMD GPU驱动 AMD驱动下载地址:https://www.amd.com/en/suppor…
-
【重磅推荐】中国车牌识别数据集(CBLPRD)：China-Balanced-License-Plate-Recognition-Dataset-330k
大家好！今天我向大家推荐一个由我创建的全新开源数据集：China-Balanced-License-Plate-Recognition-Dataset-330k。这是一个高质量、平…
-
python pytorch模型转onnx模型（多输入+动态维度）
（多输入+动态维度）整理的自定义神经网络pt转onnx过程的python代码，记录了pt文件转onnx全过程，简单的修改即可应用。 pt文件转onnx步骤 1、编写预处理代码 2、…
-
RuntimeError: expected scalar type float but found
问题描述 ————————-
-
AttributeError: module ‘torch‘ has no attribute ‘cuda‘
系统环境是Linux ，显卡：nvida-T4。 看了下原因为没有装pytorch。（印象中是装了的不知道什么时候这台服务器没有了。。） 解决方案：到pytorch官网上找到对应的…
-
Vitis-AI量化编译YOLOv5（Pytorch框架）并部署ZCU104（一）
文章目录 前言 一、Vitis-AI Pytorch框架量化（vai_q_pytorch) 二、编写量化脚本并进行量化 三、模型编译 总结 前言 https://github…
-
Linux服务器安装pytorch
安装pytorch： 出现以下字样即为成功 前提： 假设你已经安装了Python 3.9版本，并且已经正确设置了Python的环境变量。如果你的系统中有多个Python版本，请使…
-
Pytorch—模型微调（fine-tune）
随着深度学习的发展，在大模型的训练上都是在一些较大数据集上进行训练的，比如Imagenet-1k，Imagenet-11k,甚至是ImageNet-21k等。但我们…
-
PyTorch搭建卷积神经网络（CNN）进行视频行为识别（附源码和数据集）
需要数据集和源码请点赞关注收藏后评论区留下QQ邮箱~~~ 一、行为识别简介 行为识别是视频理解中的一项基础任务，它可以从视频中提取语义信息，进而可以为其他任务如行为检测，行为定位等…
-
GPU版本PyTorch详细安装教程
目录 一、安装显卡驱动 1、查看显卡驱动型号 2、下载显卡驱动 3、查看GPU状态 二、安装Visual Studio 2019 三、安装CUDA 1、下载对应版本的CUDA 2、…
-
【torch.nn.init】初始化参数方法解读
可参考：torch.nn.init – 云+社区 – 腾讯云 一. torch.nn.init.constant_(tensor, val) 1. 作…
-
DeepSpeed配置参数 – 快速上手
目录 DeepSpeed配置参数 – 快速上手 batch Size optimizer scheduler fp16 zero optimization csv mo…
