搜索结果
查询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 人评论 次浏览
