首页 > 科技

可在Windows 10 WSL上训练ML模型的TensorFlow-DirectML发布

2021-09-12 08:04:21大众软件
自去年起微软一直在其Windows子系统Linux服务Windows Subsystem for Linux(WSL)上为机器学习模型开发GPU加速训练,如今微软已经将该项目退出预览,作为TensorFlow-DirectML发布。TensorFlow-DirectML是一个分支,包括谷歌的TensorFlow机器学习平台和微软的DirectML API。DirectML为机器学习任务提供GPU加速工具,可在AMD、高通、英伟达和英特尔芯片上的DirectX 12 GPU上跨硬件和驱动程序工作。微软借助TensorFlow-DirectML提供了一个硬件加速平台,可供多个供应商访问以训练机器学习模型。用户可在GitHub找到这个开源项目,满足硬件需求即可使用。https://github.com/microsoft/tensorflow-directml