搜索结果
查询Tags标签: pytorch,共有 396条记录-
在树莓派上实现numpy的LSTM长短期记忆神经网络做图像分类,加载pytorch的模型参数,推理mnist手写数字识别
这几天又在玩树莓派,先是搞了个物联网,又在尝试在树莓派上搞一些简单的神经网络,这次搞得是LSTM识别mnist手写数字识别 训练代码在电脑上,cpu就能训练,很快的:import torch import torch.nn as nn import torchvision import numpy as np import os from PIL import…
2023/5/31 14:22:05 人评论 次浏览 -
在树莓派上使用numpy实现简单的神经网络推理,pytorch在服务器或PC上训练好模型保存成numpy格式的数据,推理在树莓派上加载模型
这几天又在玩树莓派,先是搞了个物联网,又在尝试在树莓派上搞一些简单的神经网络,这次搞得是mlp识别mnist手写数字识别 训练代码在电脑上,cpu就能训练,很快的:1 import torch2 import torch.nn as nn3 import torch.optim as optim4 from torchvision import dataset…
2023/5/30 18:22:21 人评论 次浏览 -
macbook苹果m1芯片训练机器学习、深度学习模型,resnet101在mnist手写数字识别上做加速,torch.device("mps")
apple的m1芯片比以往cpu芯片在机器学习加速上听说有15倍的提升,也就是可以使用apple mac训练深度学习pytorch模型!!!惊呆了安装apple m1芯片版本的pytorch 然后使用chatGPT生成一个resnet101的训练代码,这里注意,如果网络特别轻的话是没有加速效果的,还没有cpu的计…
2023/5/24 18:22:04 人评论 次浏览 -
蛋白质深度学习
本文主要面向两类目标读者: 一类是想使用机器学习的生物学家,一类是想进入生物学领域的机器学习研究者。如果你不熟悉生物学或机器学习,仍然欢迎你阅读本文,但有时你可能会觉得有点读不太懂!如果你已经熟悉这两者,那么你可能根本不需要本文 —— 你可以直接跳到我们的…
2023/5/11 1:22:05 人评论 次浏览 -
windows下使用pytorch进行单机多卡分布式训练
现在有四张卡,但是部署在windows10系统上,想尝试下在windows上使用单机多卡进行分布式训练,网上找了一圈硬是没找到相关的文章。以下是踩坑过程。 首先,pytorch的版本必须是大于1.7,这里使用的环境是: pytorch==1.12+cu11.6 四张4090显卡 python==3.7.6使用nn.DataP…
2023/4/2 18:22:02 人评论 次浏览 -
了解Pytorch|Get Started with PyTorch
Basics就像Tensorflow一样,我们也将继续在PyTorch中玩转Tensors。从数据(列表)中创建张量data = [[1, 2],[3, 4]]tensors = torch.tensor(data)tensorstensor([[1, 2],[3, 4]])从NumPy创建np_array = np.arange(10)tensor_np = torch.from_numpy(np_array)tensor_npten…
2022/12/4 1:23:57 人评论 次浏览 -
从 Numpy+Pytorch 到 TensorFlow JS:总结和常用平替整理
如何拥有较为平滑的移植体验?保持两种语言,和两个框架的API文档处于打开状态,并随时查阅:Python,JavaScript;Pytorch,TensorFlow JS(用浏览器 F3 搜索关键词)。可选阅读,《动手学深度学习》,掌握解决常见学习问题时,Pytorch 和 TensorFlow 的实现方法。 精读 …
2022/11/24 1:23:55 人评论 次浏览 -
Pytorch模型量化
在深度学习中,量化指的是使用更少的bit来存储原本以浮点数存储的tensor,以及使用更少的bit来完成原本以浮点数完成的计算。这么做的好处主要有如下几点:更少的模型体积,接近4倍的减少;可以更快的计算,由于更少的内存访问和更快的int8计算,可以快2~4倍。一个量化后的…
2022/10/27 1:24:56 人评论 次浏览 -
基于anaconda3的Pytorch环境搭建
方法一安装anaconda3,版本选择新的就行 打开anaconda prompt创建虚拟环境conda create -n pytorch_gpu python=3.9,pytorch_gpu是环境名称,可自行选取,python=3.9是选择的python版本,可自行选择,conda会自动下载选择的python版本 接下来去pytorch官网 https://pyt…
2022/9/12 23:25:51 人评论 次浏览 -
pytorch中的forward()的使用与解释
参考 pytorch中的forward()的使用与解释 - 云+社区 - 腾讯云前言最近在使用pytorch的时候,模型训练时,不需要使用forward,只要在实例化一个对象中传入对应的参数就可以自动调用 forward 函数 即: forward 的使用class Module(nn.Module):def __init__(self):supe…
2022/9/10 23:23:11 人评论 次浏览 -
pytorch相关知识点
1. super的作用 super()函数可以隐式的将子类里的method和父类里的method进行关联,这样就不需要再子类里显式的重新创建父类method里的属性 说人话就是继承父类需要对用到的父类的属性进行初始化,super()帮你处理完了 pytorch中的super用于初始化继承的nn.Module中的参数…
2022/9/1 23:25:33 人评论 次浏览 -
TensorFlow和CUDA、cudnn、Pytorch以及英伟达显卡对应版本对照表
TensorFlow和CUDA、cudnn、Pytorch以及英伟达显卡对应版本对照表 CUDA下载地址 CUDNN下载地址 torch下载 英伟达显卡下载 一、TensorFlow对应版本对照表版本 Python 版本 编译器 cuDNN CUDAtensorflow-2.9.0 3.7-3.108.1 11.2tensorflow-2.8.0 3.7-3.108.1 11.2tensorflow…
2022/8/27 23:52:55 人评论 次浏览 -
TensorFlow和CUDA、cudnn以及Pytorch对应版本对照表
TensorFlow和CUDA、cudnn以及Pytorch对应版本对照表 CUDA下载地址 CUDNN下载地址 torch下载 一、TensorFlow对应版本对照表版本 Python 版本 编译器 cuDNN CUDAtensorflow-2.9.0 3.7-3.108.1 11.2tensorflow-2.8.0 3.7-3.108.1 11.2tensorflow-2.7.0 3.7-3.98.1 11.2tenso…
2022/8/27 23:35:06 人评论 次浏览 -
关于安装PyTorch
1 假设你的机器上已经安装了python 和pip(没有安装的话可以参考) https://www.cnblogs.com/8335IT/p/16581772.html 2 直接安装torchvision是安装不了的,会中途报错 需要先安装pytorch 再安装torchvision 使用国内源: pip install torch -i https://pypi.tuna.tsinghu…
2022/8/15 23:56:00 人评论 次浏览 -
pip安装报错:Command python setup.py egg_info failed with error code 1
原文链接 windows下使用pip安装torch模块,出现错误:ERROR: Command "python setup.py egg_info" failed with error code 1 in C:\Users\xxx\AppData\Local\Temp\pip-install-yqzlud5w\torch\方法一 尝试参考文章:https://mp.weixin.qq.com/s/u5uIjnABGXTJGo…
2022/8/15 14:55:18 人评论 次浏览