前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布

DIGITS

作者头像
GPUS Lady
发布2018-04-02 15:31:12
9070
发布2018-04-02 15:31:12
举报
文章被收录于专栏:GPUS开发者GPUS开发者

给大家介绍一个工具,叫做DIGITS。DIGITS是一个基于WEB UI交互式的深度学习训练系统。对于深度学习的初学者或数据科学家来说,这个工具能够为他们提供极大的便利性,主要有四个方面的功能。

1、进行数据的预处理。比如说我可以把一些原始的jpeg格式的图片转成lmdb或者leveldb这样的数据库导入到DIGITS。

2、我可以利用DIGITS去配置我的网络模型,比如我去选某个Deep Learning的框架,某一个标准的模型,比如AlexNet、GoogLeNet等,或者自定义一个网络模型,然后去设置模型训练参数,以及要使用哪些GPU。

3、可以实时动态地监控整个训练过程。当导入数据和配置网络模型完成后,就可以点击按钮开始训练,在训练过程中可以实时动态地去监控loss和accuracy这样的参数曲线以及GPU、 CPU的利用率,包括显存和内存的使用情况都可以一目了然。

4、训练完成之后,我们还可以使用保存的训练好的model的参数进行一些测试,包括网络层可视化这样的工作。目前DIGITS支持的深度学习框架包括了Caffe、PyTorch和TensorFlow,支持的算法主要是包含了图像分类、图像分割、目标检测。

下面展示几组Tesla V100在不同的Deep Learning的framework上,针对不同的网络模型的一些benchmark测试结果,他们对比的都是八颗Tesla V100以及八颗P100的性能加速。左边展示的是Tesla V100的PCIe卡的性能加速,右边展示的是Tesla V100的NVLink卡的性能加速。针对不同的framework,实际上Tesla V100的加速效果确实会略有不同,需要说明的是,这些framework测试中包括Caffe、Caffe2、TensorFlow、MXNet以及PyTorch,他们都支持Tensor Core,但是目前CNTK还不支持Tensor Core。

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2018-01-27,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 GPUS开发者 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档