首页>科技动态 > 科技前沿 > 正文

TensorFlow-DirectML正式版:可为WSL模型训练实现GPU加速

微软发布了 TensorFlow-DirectML 的正式版,该工具可让用户在 Windows 的 Linux 子系统(WSL)上 GPU 加速模型训练。

顾名思义,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、3.6 或 3.7;

以下支持的 GPU 之一:

AMD Radeon R5/R7/R9 2xx 系列或更新版本;

英特尔 HD Graphics 5xx 或更新版本;

英伟达 GeForce GTX 9xx 系列 GPU 或更新版本。

标签: 微软 WSL GPU

推荐DIY文章
性能旗舰一加 10 Pro首销战报出炉 1秒破亿
一加 10 Pro重磅发布,最强性能旗舰10至名归
千万销量千家售后 一加多触点布局国内市场
科技加持,未来已来,“头号玩家”不再是科幻
一加 10 Pro发布在即,性能屏幕影像全方位提升
一加中国区新任总裁李杰 :做好品牌与用户之间的连接器
精彩新闻

超前放送