jittor 和pytorch gpu 使用效率对比（惊人jittor的算力利用率是pytorch 4-5倍）

之前使用的是cpu对比 pytorch 好像更胜一筹（本人觉得是当时可能环境不对这次配置好了完美环境使用lsgan代码进行对比果然如jittor官网所说比pytorch快，但是本人还是有一个惊奇的发现使用的显存还比pytorch少也就是说我们同样的网络我们可以训练比pytorch batchsize大一倍的数据且速度是pytorch 2.7倍）

不信的小伙伴可以测试一下
pytorch 结果
使用gpu显存1777m使用率100%
使用时间
388

[Epoch 0/200] [Batch 937/938] [D loss: 0.244307] [G loss: 0.249894]
388.5762474536896
[Epoch 1/200] [Batch 0/938] [D loss: 0.247776] [G loss: 0.255582]

jittor结果
使用gpu显存1004m 使用率94%
使用时间152

[Epoch 0/200] [Batch 937/938] [D loss: 0.218945] [G loss: 0.421059]
152.86592602729797
[Epoch 1/200] [Batch 0/938] [D loss: 0.210003] [G loss: 0.288427]

jittor代码


import jittor as jt
from jittor import init
from jittor

jittor 和pytorch gpu 使用效率对比（惊人jittor的算力利用率是pytorch 4-5倍）相关推荐

PyTorch GPU环境搭建【Winds10 + Python3.6 + CUDA10.0 + PyTorch 0.4.1 (PyTroch 1.0 branch)】
以下链接是个人关于深度学习环境搭建的所有链接,包含了各个框架: 深度学习环境搭建-史上最全无死角系列有兴趣的朋友可以添加微信 17575010159 相互交流,有交流群. 前言首先要安装ancon ...
Pytorch gpu加速方法
Pytorch gpu加速方法原文: https://www.zhihu.com/question/274635237 relu 用 inplace=True 用 eval() 和 with tor ...
pytorch GPU版本安装 win10
我原先装的是pytorch cpu版本,在复现论文源码的时候提示AssertionError: Torch not compiled with CUDA enabled,查了一下资料,又看了一下电脑的 ...
gpu超算算法_超算安装GPU-based软件 (以pytorch为例)
一般的超算的拓扑结构是若干个登陆节点+若干个交换机+大量计算CPU节点+大量GPU计算节点+一个(或若干个)存储节点+管理节点. 其中存储节点的共享存储可以被所有节点访问. 一般运作方式是,如果我的计 ...
Win11-GTX3060-配置Pytorch GPU
文章目录所需清单 1.安装CUDA11.3 2.安装cuDNN 11.x 3.安装Anaconda 4.打开Anaconda Prompt安装配置Python与Pytorch环境 5.Jupyter ...
Pytorch(GPU)配环境原理：cuda+cudnn+pytorch配环境的每一步到底干了些什么？
作者:18届cyl 时间:2022.5.11 参考文章:https://blog.csdn.net/qq_42406643/article/details/109545766 最近帮舍友配pytorc ...
Windows下命令行安装Pytorch(GPU)
对于一个小白来说,找了好多博客,都是告诉你去下载cuda,cudnn,然后安装配置各种环境,十分麻烦.最近才发现了最简单的安装方法,只需要在cmd中输入简单的几行命令即可. 注意:以下命令行要在对应的 ...
安装pytorch gpu版本
在windows下用anaconda虚拟环境安装pytorch gpu版成功! 文章目录 0. 用pip安装 1. 安装准备 2. 安装指令 3. 查看.提升cuda版本 3.1 判断显卡是否支持c ...
深度学习环境配置超详细教程【Anaconda+PyTorch(GPU版)+CUDA+cuDNN】
深度学习环境配置入门深度学习,首先要做的事情就是要搭建深度学习的环境.不管你是Windows用户,Mac用户还是Ubuntu用户,只要电脑配置允许,都可以做深度学习,毕竟Windows.Mac和Ub ...

jittor 和pytorch gpu 使用效率对比（惊人jittor的算力利用率是pytorch 4-5倍）

jittor 和pytorch gpu 使用效率对比（惊人jittor的算力利用率是pytorch 4-5倍）相关推荐

最新文章

热门文章