0
点赞
收藏
分享

微信扫一扫

tensorflow/pytorch/mindspore在VGG16前向传播上的性能对比

首先说下mindspore,作为华为的主打软件产品,该计算框架可用性一直较差,不同版本不同计算硬件下的代码往往都不是完全兼容的,也就是说你在mindspore的官网上找到的VGG预训练模型的代码是mindspore-1.5.0的版本在ascend平台的,如果你用mindspore-gpu-1.9.0的版本就无法运行,这是mindspore最常见的错误问题,因此mindspore在VGG16前向传播的性能表现为零,因为根本就无法运行官方的源代码,除非你手动去修改mindspore的框架代码,这事情我去年是会这么干的,今年我说啥也不会干了。如果你给一个开源的产品提了N多的意见,结果过好久一看,啥变化没有,如果非说有变化那就是变得越来越差了,那估计换谁也都不会支持了。

 

 

TensorFlow这里使用的是1.14.0版本,由于TensorFlow在1.x和2.x之间上的巨大差异,以及2.x版本的较差使用性,因此这里使用的是TensorFlow的作为经典的版本1.14.0,当然并没有使用TensorFlow的最新版本来比较对于TensorFlow来说并不很公平,但是TensorFlow2.x版本的相关代码和技术博客基本已经从互联网上绝迹,作为一个几乎要濒临灭绝的计算框架在这里使用旧版本也是实属无奈。相关代码参考:

​​【转载】 Tensorflow如何直接使用预训练模型(vgg16为例) ​​

 

tensorflow/pytorch/mindspore在VGG16前向传播上的性能对比_深度学习

 

 

 

 

Pytorch

运行版本gpu1.13.0:

import torch
from torchvision import datasets, models, transforms
import numpy as np

# with torch.no_grad():
# model_ft = models.vgg16(pretrained=True).to("cuda:1")
model_ft = models.vgg16(pretrained=True).to("cuda:1")

batchsize=32

import time
a_time = time.time()
for _ in range(500):
images = np.random.random(batchsize*224*224*3).reshape(batchsize,3,224,224)
model_ft(torch.tensor(images, dtype=torch.float32, device='cuda:1'))

print(time.time()-a_time)

View Code

tensorflow/pytorch/mindspore在VGG16前向传播上的性能对比_杂谈_02

 

 

 

=====================================

 

 

运行硬件平台:

CPU: 5.0Ghz下的因特尔10700k

GPU: RTX2070SUPER

 

 

其他参数:

batch_size=32

迭代次数:500

 

 

对比结果:

mindspore-gpu-1.9.0版本的性能表现:0分(根本无法运行)

TensorFlow-gpu-1.14.0版本的性能表现:用时170秒

Pytorch-gpu-1.13.0版本的性能表现:用时42秒

 

 

 

 

================================================

 

 

PS: 可以看到在现如今的深度学习框架问题上,不论是易用性还是整体性能上pytorch都已经成为了当之无愧的NO.1,虽然TensorFlow也还能用,不过已经在多个方面上被pytorch超越了,如果非说pytorch有不如TensorFlow的地方那估计也就是曾经的历史遗留的项目了,而至于mindspore的gpu版本这里基本可以认为是不可用的状态,当然mindspore框架的主要定位目标平台是ascend平台,所以或许mindspore在ascend平台上会有截然不同的表现也是不可知晓的。

 

 

---------------------------------------------------

 

举报

相关推荐

0 条评论