您当前的位置:快讯网 > 财经

微软TensorFlow-DirectML正式版发布:可在WSL上GPU

2021-09-13 12:53 来源:IT之家 作者:顾晓芸 阅读量:7581 

,微软最近几天发布了TensorFlow的官方版本——DirectML,让用户可以在Windows Linux子系统上训练GPU加速机器学习模型。

顾名思义,TensorFlow—DirectML就是利用Google TensorFlow机器学习平台适配微软DirectML机器学习API,从而在Win10/WSL上训练机器学习模型时支持GPU加速。

IT之家了解到,用户可以在Python环境下输入pip install tensorflow—directml,支持在各种DirectX 12兼容硬件上进行复杂机器学习模型的训练和推理。

Win10软件和硬件要求:

Windows 10 1709,64位,

Python x86—64 3.5,3.6或3.7,

以下受支持的图形处理器之一:

AMD R5/R7/R9 2xx系列或更高版本,

高清显卡5xx或更高版本,

英伟达GeForce GTX 9xx系列GPU或更高版本,

WSL硬件和软件要求:

Windows 10内部预览版,64位,

Python x86—64 3.5,3.6或3.7,

以下受支持的图形处理器之一:

AMD R5/R7/R9 2xx系列或更高版本,以及20.20.01.05 WSL驱动程序,

高清显卡6xx或更高版本,以及28.20.100.8322 WSL驱动程序,

英伟达GeForce GTX 9xx系列或更高版本,以及460.20 WSL驱动程序,

具体用法可以查看TensorFlow—DirectML的GitHub页面。

本文地址:http://www.chinaxhk.net/finance/12213.html - 转载请保留原文链接。
免责声明:本文转载上述内容出于传递更多信息之目的,不代表本网的观点和立场,故本网对其真实性不负责,也不构成任何其他建议;本网站图片,文字之类版权申明,因为网站可以由注册用户自行上传图片或文字,本网站无法鉴别所上传图片或文字的知识版权,如果侵犯,请及时通知我们,本网站将在第一时间及时删除。

热门推荐
返回顶部