近期,微软发布了一款名为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命令,以及排程工作。
责任编辑:pj
-
微软
+关注
关注
4文章
6723浏览量
107390 -
gpu
+关注
关注
28文章
5110浏览量
134512 -
Linux
+关注
关注
88文章
11635浏览量
218104
发布评论请先 登录
Hbirdv2 OpenOCD的编译
SkyOne® Ultra 2.0 前端模块,适用于 WCDMA / LTE 频段 1、2、3、4、34、39 skyworksinc
WSL 1 和 WSL 2 的区别是什么
适用于 WLAN 和蓝牙®应用的 2.4 GHz 高效前端 skyworksinc
龙芯处理器支持WINDOWS吗?
聚徽厂家解码——工控机操作系统选择:Windows、Linux、QNX 如何匹配工业场景
是否提供适用于CY7C65213 的 Windows 11驱动程序和 API 库 (dll)?
Skyworks ICE™ Technology 2 GHz 高功率前端模块,适用于 Wi-Fi 7 应用 skyworksinc
如何将FX3与WSL(Linux 的 Windows 子系统)一起使用?
适用于Oracle的Python连接器:可访问托管以及非托管的数据库
适用于Oracle的ODBC驱动程序
风华高压贴片电容器适用于哪些应用场景?

微软研发适用于Linux的Windows子系统——WSL 2
评论