天天看点

AMD深度学习库MIOpen更新,支持CNN加速

本文来自AI新媒体量子位(QbitAI)

农企的深度学习加速库MIOpen 1.0更新了,它现在已经能支持对CNN的加速。

ROCm全称Radeon Open Compute platform,是AMD在去年12月推出的开源GPU运算平台,MIOpen则是AMD为此开发的软件库,其作用是将程序设计语言和ROCm平台连接,以充分利用GCN架构。

此次更新包括:

同时为正向和反向传播优化的深度卷积解算器

优化的卷积算法,包括Winograd和快速傅立叶转换

为深度学习准备的、优化的GEMM

池化、Softmax、激活、梯度算法的批量归一化,以及LR Normalization

4D张量NCHW格式

支持OpenCL和HIP的框架API

对MIOpen驱动的支持,以测试任何正向/反向网络

支持Ubuntu 16.04和Fedora 24的二进制封装

ROCm支持如下深度学习平台:

AMD深度学习库MIOpen更新,支持CNN加速

是的,没有Theano。

源代码:

https://github.com/ROCmSoftwarePlatform/MIOpen

文档:

MIOpen:

https://rocmsoftwareplatform.github.io/MIOpen/doc/html/apireference.html

MIOpenGemm:

https://rocmsoftwareplatform.github.io/MIOpenGEMM/doc/html/index.html

【完】

本文作者:吴唯

原文发布时间:2017-07-03

继续阅读