ROCm:AMD系开源HPC/超规模GPU计算/深度学习平台

ROCm的英文全称Radeon Open Compute platform,它是AMD在去年12月推出的一款开源GPU运算平台,目前已经发展到了1.3版本。MIOpen则是AMD为此开发的软件库,其作用是将程序设计语言和ROCm平台连接,以充分利用GCN架构。

本次发布的版本包括以下内容:

  • 深度卷积解算器针对前向和后向传播进行了优化
  • 包括Winograd和FFT转换在内的优化卷积
  • 优化GEMM深入学习
  • Pooling、Softmax、Activations、梯度算法的批量标准化和LR规范化
  • MIOpen将数据描述为4-D张量 - Tensors 4D NCHW格式
  • 支持OpenCL和HIP的框架API
  • MIOpen驱动程序可以测试MIOpen中任何特定图层的向前/向后网络
  • 二进制包增加了对Ubuntu 16.04和Fedora 24的支持
  • 源代码位于https://github.com/ROCmSoftwarePlatform/MIOpen
  • 参考文档

ROCm 1.6具有MIOpen的预构建软件包

安装ROCm MIOpen实现(这是建立在已经安装了“ROCm”和“ROCm-opencldev”包的前提下):

仅支持OpenCL开发

sudo apt-get install miopengemm miopen-opencl

仅支持HIP开发

sudo apt-get install miopengemm miopen-hip

或者你可以从源代码中构建

深度学习框架增加了对ROCm的支持

参考链接


ROCm:AMD系开源HPC/超规模GPU计算/深度学习平台

发布者

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注