微软推出TensorFlow扩展套件DirectML 支持DirectX 12

  近期，微软发布了一款名为TensorFlow-DirectML的TensorFlow扩展套件，帮助开发者能够在Windows平台上通过支持DirectX 12的GPU来加速TensorFlow模型运算，并同样适用Linux的Windows子系统(WSL)。目前该套件已经在GitHub开源，供开发者取用程序码。

  此前微软便在build 2020大会上曾宣布，适用于Linux的Windows子系统WSL 2(Windows Subsystem for Linux 2)开始支持GPU硬件加速。但微软表示，TensorFlow虽可以同时利用CPU和GPU来执行运算，但GPU加速仅限于特定供应商的平台。

  另一方面DirectX一直以来，被广泛地用于加速Windows上的游戏和计算应用程序，其DirectML可以让支援DirectX 12的GPU，执行高效能的数学运算。因此微软透过整合TensorFlow以及DirectML，让开发者可以在Windows，使用GPU执行完整的机器学习训练和推理功能。

  TensorFlow-DirectML是通过加入名为DML而非GPU的新装置，将TensorFlow和DirectML整合在一起，其新的核心集建构在DirectML API之上，而非象是现有的CPU和GPU核心，建立在Eigen函式库之上。

  由于DirectML又是以Direct3D 12低阶函式库开发，该绘图API是专为需要对资源分配和工作排程，进行绝对控制的高效能低延迟应用程序设计，因此DirectML与TensorFlow整合，还需要仰赖装置Runtime负责在装置与主机间，管理装置存储器、复制张量、纪录GPU命令，以及排程工作。