本文来自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支持如下深度学习平台:
是的,没有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