您尚未登录账户请先登录您的atyun账户，方可使用该功能
您的企业账号申请正在审核中审核通过后即可使用此功能，请耐心等待~
AMD的ROCm GPU现已支持TensorFlow
2018年08月29日 由 浅浅 发表
43714
0
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
广告
广告
写评论取消
回复取消