Jittor 深度学习框架入门(pytorch转换)、对比

jittor 和pytorch gpu 使用效率对比(惊人jittor的算力利用率是pytorch 4-5倍) 之前使用的是cpu对比 pytorch 好像更胜一筹(本人觉得是当时可能环境不对这次配置好了完美环境使用lsgan代码进行对比果然如jittor官网所说比pytorch快,但是本人还是有一个惊奇的发现使用的显存还比pytorch少也就是说我们同样的网络我们可以训练比pytorch batchsize大一倍的数据且速度是pytorch 2.7倍) 阅读详情

1.PyTorch是一个开源的Python机器学习库,基于Torch,用于自然语言处理等应用程序。

中文网站:https://pytorch-cn.readthedocs.io/zh/latest/

2.NumPy(Numerical Python)是Python的一种开源的数值计算扩展。这种工具可用来存储和处理大型矩阵,比Python自身的嵌套列表(nested list structure)结构要高效的多(该结构也可以用来表示矩阵(matrix)),支持大量的维度数组与矩阵运算,此外也针对数组运算提供大量的数学函数库。

3.tensorflow

tensorFlow是一个基于数据流编程(dataflow programming)的符号数学系统,被广泛应用于各类机器学习(machine learning)算法的编程实现,其前身是谷歌的神经网络算法库DistBelief。

tensorflow和pytorch都基于caffe而来,并且日趋同化

Caffe,全称Convolutional Architecture for Fast Feature Embedding,是一个兼具表达性、速度和思维模块化的深度学习框架。由伯克利人工智能研究小组和伯克利视觉和学习中心开发。虽然其内核是用C++编写的,但Caffe有Python和Matlab 相关接口。Caffe支持多种类型的深度学习架构,面向图像分类和图像分割,还支持CNN、RCNN、LSTM和全连接神经网络设计。Caffe支持基于GPU和CPU的加速计算内核库,如NVIDIA cuDNN和Intel MKL。

对比:

pytorch 轻量、灵活,使用动态图描述神经网络

动态图分为静态的子图和控制流,控制流控制宏观上调用哪些子图,静态包含Module和Function,Function包含自动微分相关管理,Module在此基础上管理权重。

控制流使用python 的控制流,在Module的forward中使用python描述控制流程。

tensorflow早期使用静态图描述神经网络,但缺乏控制流。目前是加一些控制流的op(用静态图打一些补丁把动态的部分包装成一个op,但不灵活。用户层面使用较多的是python描述控制流。

pytorch的做法是引入Torchscript,用python修饰符将用户描述控制流转为Torchscript控制流,同时转换为后端,以便部署。

4.jittor

Jittor 是一个基于即时编译和元算子的高性能深度学习框架,整个框架在即时编译的同时,还集成了强大的Op编译器和调优器,为您的模型生成定制化的高性能代码。

Jittor前端语言为Python。前端使用了模块化的设计,类似于PyTorch,Keras,后端则使用高性能语言编写,如CUDA,C++。

https://cg.cs.tsinghua.edu.cn/jittor/assets/docs/jittor.html

pytorch->jittor模型辅助转换工具教程

http://cg.cs.tsinghua.edu.cn/jittor/tutorial/2020-5-2-16-43-pytorchconvert/

from jittor.utils.pytorch_converter import convert

pytorch_code=
"""
class AdaptiveAvgPool2d(nn.Module):

"""
jittor_code = convert(pytorch_code)
print(jittor_code)

在线转换:

http://cg.cs.tsinghua.edu.cn/jittor/pt_converter/

pytorch和jittor 程序对比 对比结果 pytorch 999 使用时间0.59 loss 0.000130 jittor 999 使用时间 1.009 loss 0.00037 结论很明显 阅读详情

相关推荐

jittor和pytorch生成网络对比之stargan

jittor代码 import glob import random import os import numpy as np from jittor.dataset.dataset import Dataset import jittor.transform as transform from PIL import Image class CelebADataset(Dataset): def __init__(self, root, transform_=None, mode="train"

东方佑 673

Jittor(计图)深度学习框架安装

Jittor(计图)深度学习框架安装

weixin_43955191的博客 7989

静态与动态LACP链路聚合:原理、配置与排错实战指南

链路聚合技术是提升网络带宽与可靠性的核心方案,通过将多个物理端口逻辑捆绑,实现带宽叠加与链路冗余。其原理基于IEEE 802.3ad标准,通过负载均衡算法将数据流分布到不同物理链路,既避免了数据包乱序,又提升了整体吞吐量。该技术的核心价值在于消除单端口瓶颈、提供毫秒级故障切换,并能简化网络拓扑管理。在企业服务器接入、数据中心互联及虚拟化平台等场景中,链路聚合已成为构建高可用网络的基础。实践中,动态LACP模式通过协议报文协商,能主动检测链路故障并防止配置错误导致的网络环路,相比静态聚合更为可靠。本文将深入解

weixin_31412781的博客 267

【亲测免费】 Jittor:快速、灵活的深度学习框架

是一个专为深度学习设计的开源框架,它以其高效的计算性能和灵活的编程模型吸引着越来越多的开发者。本文将深入探讨Jittor的特点、技术优势以及应用场景,帮助您更好地理解并利用这一工具。 ## 项目简介 Jittor是一个基于Just In Time (JIT) 编译器的深度学习框架,旨在提供与NumPy类似的API,但其核心计算是在GPU上进行的,以实现更快的速度。通过动态图模式和即时编译,Ji...

gitblog_00041的博客 1149

【WorkBuddy】WorkBuddy 智能体快速上手:从零到一的实践指南

文章摘要: 传统自动化脚本难以应对灵活多变的自然语言需求,导致逻辑复杂、维护成本高。智能体(Agent)通过理解语义意图、动态决策和松耦合组件解决了这些问题。WorkBuddy采用声明式配置,通过可视化界面定义意图、实体和对话策略,无需编码即可快速构建智能体。其核心组件包括意图分类、实体提取、对话策略、故事样本和回退机制,适用于客服支持、知识问答等高频场景。通过环境准备(Python/Node.js/Redis)和简单配置,用户可以快速搭建可运行的智能体,显著提升自动化效率。

ruiqideng的博客 1631

清华开源Jittor:首个国内高校自研深度学习框架,一键转换PyTorch

点击“开发者技术前线”,选择“星标????”在看|星标|留言, 真爱机器之心编辑部继 Theano、Caffe 之后,又一个由高校主导的深度学习框架开...

开发者技术前线-DevolperFront 631

软硬件协同设计的大小端问题

缘起 最近需要在soc系统内做协处理器,加速AES加密运算,寄存器一次读取32bit,AES一次处理128bit,因此在数据拼接的时候,需要考虑大小端的问题,特来记录一番。 一、什么是大小端 大小端的问题发生在寄存器和内存间传递数据,这点非常重要,很多地方简单地说是数据存储方式,其实不准确。追根溯源,大小端问题是在设计load,store指令的时候考虑的,因此本质是寄存器和内存交互的问题,其他地方不会发生。 比如c语言里定义一个变量int a=ox1234,那么查看在线汇编平台的对应汇编代码: # 给2号寄

http://www.icfgblog.com/ 613

Jittor 与 PyTorch 双框架环境隔离:Conda 独立部署 2 套深度学习环境

本文详细介绍了如何在Ubuntu系统中使用Conda实现Jittor与PyTorch双框架环境的独立部署与隔离管理。通过创建两个完全隔离的Conda环境,开发者可以避免CUDA版本冲突、Python版本差异等常见问题,实现高效的多框架开发。文章包含环境配置、框架安装、性能优化及常见问题解决方案,助力深度学习开发者提升工作效率。

weixin_34099526的博客 482

3DGS-to-PC实战指南:零代码实现3D高斯场景到点云/Mesh的高效转换

本文详细介绍了3DGS-to-PC工具如何实现3D高斯场景到点云/Mesh的高效转换,无需代码即可完成。通过实战案例和参数调优技巧,帮助用户快速掌握这一开源插件的使用方法,提升3D数据处理效率。特别适合建筑扫描、影视特效等领域的专业人士。

q1w3e5r7t9y的博客 273

Python开发者必看:5分钟搞定Gemini API的3种调用方式(含代码示例)

本文为Python开发者详细介绍了三种高效调用Gemini API的核心方式:基础同步调用、流式调用和聊天模式,并涵盖多模态应用等高级特性。通过清晰的代码示例和实战场景分析,帮助开发者快速掌握Gemini API的集成与优化,提升AI应用开发效率。

g8f9d0s1a2的博客 1122

Jittor与PyTorch对比分析:性能、易用性和生态系统全面评测

Jittor作为国内自主研发的高性能深度学习框架,在JIT编译和元算子技术方面展现出独特优势。本文将从性能表现、API易用性和生态系统建设三个维度,对Jittor与PyTorch进行深度对比分析,为深度学习开发者的框架选择提供参考依据。 ## 性能表现:JIT编译带来的计算效率提升 Jittor框架基于即时编译技术,能够将Python代码动态编译为高性能的机器码。在内存带宽测试中,Jittor

gitblog_00802的博客 659

AI工具赋能日常工作的实践指南:以产品经理与程序员为例

AI工具赋能日常工作的实践指南:以产品经理与程序员为例

专注于编程、AI、人工智能等多个领域内容研究和分析 591

清华开源 Jittor:首个国内高校自研深度学习框架,一键转换 PyTorch

点击上方“视学算法”,马上关注真爱,请设置“星标”或点个“在看”来源 |机器之心继 Theano、Caffe 之后,又一个由高校主导的深度学习框架开源...

zandaoguang的博客 780

【用unity实现100个游戏之16】Unity程序化生成随机2D地牢游戏1(附项目源码)

关于使用TileMap生成随机2D地图,其实之前已经有做过类似的,感兴趣可以看看 但是随着学习深入,发现之前做的比较粗糙和不够全面,最近又在外网看到

向宇的博客,专注php/web全栈 unity游戏开发,欢迎大家评论纠错 2215

STM32F407中文手册

STM32f407中文手册 本参考手册面向应用开发人员,提供有关使用 STM32F405xx/07xx、STM32F415xx/17xx、 STM32F42xxx 和 STM32F43xxx 微控制器存储器与外设的完整信息。STM32F405xx/07xx、STM32F415xx/17xx、STM32F42xxx 和 STM32F43xxx 构成一个微 控制器系列,各产品具有不同的存储器大小、封装和外设。有关订购信息以及器件的机械与电气特性,请参见数据手册。有关 ARM Cortex™-M4F 内核的信息,请参见《Cortex™-M4F 技术参考手册》。

对标Theano、Caffe,清华开源首个国内高校自研深度学习框架——Jittor

对标Theano、Caffe,清华开源首个国内高校自研深度学习框架——Jittor 近日,清华大学计算机系图形实验室宣布开源一个全新的深度学习框架:Jittor,中文名计图,对标Theano、Caffe。 据悉,计图(Jittor:Just in Time)是一个采用元算子表达神经网络计算单元、完全基于动态编译(Just-in-Time)的深度学习框架,元算子和Numpy一样易于使用,并且超越Nu...

weixin_43266521的博客 405

OV9734 DATASHEET,完整规格书

OV9734完整规格书,包括电气参数,尺寸规格,寄存器配置表等

COTR项目迁移1:对pytorch和jittor的基本认识和配置基本环境

将一个项目从pytorch迁移到jittor的第一步:配置基本环境

qq_74959300的博客 627

汉书HE8,好用的汇编代码编写工具

汉书HE8,好用的汇编代码编写工具早期编写汇编代码所用功能强大

国产深度学习框架迎来高光时刻,继清华 Jittor开源后,旷视「天元」纷纷重磅开源!...

关注上方“深度学习技术前沿”,选择“星标公众号”,资源干货,第一时间送达!1956年,美国的达特茅斯会议成为人工智能开启的标志性事件。同年,中国第一批自主制造的汽车——解放牌汽车在长春第...

深度学习技术前沿 556

清华开源深度学习框架计图,开源超级玩家再进阶

2020-03-22 09:24 导语:清华开源计图,背后是三代人的共同努力。 雷锋网AI源创评论报道,据官方消息,清华大学计算机系图形实验室宣布开源一个全新的深度学习框架:Jittor,中文名计图。 Jittor 是一个采用元算子表达神经网络计算单元、完全基于动态编译(Just-in-Time)的深度学习框架。[1] 据介绍,Jittor 内部使用创新的元算子和统一计算图的深度学习框...

喜欢打酱油的老鸟 1033

GAN(论文+Jittor框架实现代码)

gan.py import jittor as jt from jittor import init from jittor import nn from jittor.dataset.mnist import MNIST from jittor.dataset.dataset import ImageFolder import jittor.transform as transform import argparse import os import numpy as np from loguru i

qq_41048761的博客 1353

清华「计图」、旷视「天元」纷纷重磅开源,国产深度学习框架迎来高光时刻...

来源:优快云本文约3141字,建议阅读7分钟。本文介绍清华开源的深度学习框架 Jittor(计图)引起了广泛关注,支撑旷视诸多业务和战绩的深度学习框...

zandaoguang的博客 412

Jetson Nano(4)—框架学习(4.1)

1、主流深度学习框架扫盲 万事具备,硬件和系统环境基本已经具备,目前的深度学习框架有下面的的,这里特别提一下国产的深度学习框架:华为 MindSpore、旷视 MegEngine、清华 Jittor(毕竟我是爱国的),后期有时间可以去看看。 ![在这里插入图片描述](https://img-blog.csdnimg.cn/20210624004220979.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text

记录一些平时工作或者学习中的瞬间 852

Jittor 与 Conda 环境深度集成:解决 g++ 版本冲突与 GLIBCXX 缺失 2 大难题

本文详细介绍了如何在Conda环境中深度集成Jittor框架,解决g++版本冲突与GLIBCXX缺失两大难题。通过Ubuntu系统下的实战案例,展示了从环境准备、CUDA集成到性能优化的完整流程,帮助开发者构建稳定的Jittor开发环境,特别适用于深度学习项目部署。

weixin_34008784的博客 290
上一篇: latex 多个参考文献引用,想要使用破折号
下一篇: 计算机视觉方向相关前言知识了解
竹篓有个天
博客等级 码龄9年 6431粉丝 88原创
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付:10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值