微软深度学习框架TensorFlowDirectML获AMD全力支持
人工智能是当下最热门的科学应用领域,但是相对于谷歌来说,微软在这方面起步相对较晚,这是因为几乎所有的深度学习框架在最初都只有 Linux 版本,而微软这边的主力操作系统只有 Windows。
好在随着 Windows 这边在深度学习计算加速方面日渐成熟,终于拿出了名为 DirectML 或者说 WinML 的加速标准后,Windows 开始重新找回感觉,在本月 9 日,微软发布了基于 DirectML 运行于 WSL 的 TensorFlow-DirectML 开源深度学习框架正式版。
按照微软的说法,TensorFlow-DirectML 的安装配置非常简单,在 Python 里执行 pip install tensorflow-directml 就搞定了,值得一提的是,在 TensorFlow-DirectML 的整个研发过程中,微软都和 AMD 和 NVIDIA 保持着密切的互动,确保它们的 DX12 GPU 能提供硬件训练加速体验。
AMD 这边在其官方 Blog 官宣了对 TensorFlow-DirectML 的支持,并且第一时间提供了一些 RDNA2 GPU 的性能测试数据:
实现 TensorFlow-DirectML 支持不管是对 AMD 还是 NVIDIA 来说都是绝对的好事,要知道 TensorFlow 或者锁其他深度学习框架在 Linux 里都有一个麻烦的问题,那就是版本匹配和环境配置都是对许多新手来说都是繁琐甚至噩梦,以至于各种容器大为流行。
如今,作为一个兼容性极为友好的 Windows 也提供了相近环境支持,对开发人员特别是新手来说,都是非常有意义的。
对 GPU 厂商来说,受益最大也许是 AMD,要知道,AMD 这边还在为 ROCm 的认受性而苦苦挣扎,如今 Windows 这边掉落了一个重要的装备,对 AMD 来说显然是不可多得。
最后,我想对一些对 Windows 不屑一顾的朋友们说一下,Windows 在今天依然是最为重要的图形操作系统,上面有大量的专业应用都有采用深度学习加速或者能从深度学习加速中受益,例如 Adobe 的图片处理、视频剪辑工具等,都早已引入了深度学习加速并且让用户大为受益,它是一个不可忽视的市场。