,微软最近几天发布了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 - 转载请保留原文链接。免责声明:本文转载上述内容出于传递更多信息之目的,不代表本网的观点和立场,故本网对其真实性不负责,也不构成任何其他建议;本网站图片,文字之类版权申明,因为网站可以由注册用户自行上传图片或文字,本网站无法鉴别所上传图片或文字的知识版权,如果侵犯,请及时通知我们,本网站将在第一时间及时删除。 |