pytorch
-
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…
-
01 Pytorch和CUDA对应的版本及Pytorch和Python对应的版本及Python与Anaconda的对应关系
官方推荐的cuda版本为10.2和11.3，这两种 cuda 支持大多数的 pytorch 版本。 以下是Pytorch和CUDA对应的版本 CUDA 环境 PyTorch 版…
-
已安装Pytorch，却提示no moudle named ‘torch’（没有名称为torch的模块）
Pytorch 安装过很多遍（自己的老电脑，实验室的电脑，实验室换的新电脑，服务器的……）不同的显卡不同的服务器……遇到过很多坑都解决过，本以为 Pytorch 配置环境的坑已经被…
-
No matching distribution found for torch==1.10.1+cu111
目录 注意问题 PyTorch 安装 gpu版本的问题 我的电脑配置 不想看我的经历可以直接使用下方命令 对应1.10版本的torch 对应1.9.1版本的torch 对应1.6….
-
GPU版本的pytorch安装（显卡为3060ti，如何选择对应的cuda版本）
文章目录 前言 一、基本知识 二、安装步骤 1.首先判断自己有无英伟达的显卡 2.安装或者查看自己的显卡驱动 3.显卡的算力必须与Cuda runtime version相匹配 4…
-
使用YOLOv5实现人脸口罩佩戴检测（详细）
前言 最近把李沐大神《动手学深度学习v2》的目标检测部分学完了，就想找一个项目练练手，学以致用嘛，觉着人脸口罩佩戴检测比较符合当下时代背景，所以就选择了这个项目，也是前几天刚刚完成…
