AMD的ROCm GPU现已支持TensorFlow

2018年08月29日 由 浅浅 发表 222308 0
AMD的ROCm GPU现已支持TensorFlow

AMD宣布推出支持TensorFlow v1.8的ROCm GPU,其中包括Radeon Instinct MI25。这是AMD加速深度学习的一项重要里程碑。

ROCm即Radeon Open Ecosystem,是在Linux上进行GPU计算的开源软件基础。AMD的TensorFlow实现利用了MIOpen,这是一个用于深度学习的高度优化的GPU例程库。

AMD提供了一个预构建的whl软件包,安装简单,类似于安装通用TensorFlow for Linux。AMD已经发布了安装说明(gpuopen.com/rocm-tensorflow-1-8-release/),还有预先构建的Docker镜像(hub.docker.com/r/rocm/tensorflow/)。

除了支持TensorFlow v1.8之外,AMD还在努力向TensorFlow主存储库进行所有针对ROCm的增强功能。其中一些补丁已经在上游合并,还有一些补丁正在积极审查中。在努力完全增强功能的同时,AMD将发布和维护未来支持ROCm的TensorFlow版本,例如v1.10。

AMD相信深度学习优化,可移植性和可扩展性的未来源于特定领域的编译器。AMD受到XLA早期结果的启发,还在致力于发展AMD GPU启用和优化XLA。

有关AMD在该领域工作的更多信息:www.amd.com/deeplearning
欢迎关注ATYUN官方公众号
商务合作及内容投稿请联系邮箱:bd@atyun.com
评论 登录
写评论取消
回复取消