微比恩 > 信息聚合 > 微软 TensorFlow-DirectML 正式版发布:可在 WSL 上 GPU 加速机器学习

微软 TensorFlow-DirectML 正式版发布:可在 WSL 上 GPU 加速机器学习

2021-09-12 09:32:17来源: IT之家

IT之家 9 月 12 日消息 微软近日发布了 TensorFlow-DirectML 的正式版,该工具可让用户在 Windows 的 Linux 子系统(WSL)上 GPU 加速机器学习(ML)模型训练。顾名思义,TensorFlow-DirectML 用于将谷歌 TensorFlow 机器学习平台与微软 DirectML 机器学习 API 适配,以支持在 Win10/WSL 上进行机器学习模型训练时获得 GPU 加速。IT之家了解到,用户可以在 Python 环境下(Python 3.8 及以上暂不支持)输入 pip install tensorflow-directml进行安装,支持在各种 DirectX 12 兼容硬件上训练和推理复杂的机器学习模型。Win10 软硬件要求:Windows 10 1709,64 位(Build 16299 或更高);Python x86-64 3.5

关注公众号
标签: 微软 GPU