,微软最近几天发布了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页面。
郑重声明:此文内容为本网站转载企业宣传资讯,目的在于传播更多信息,与本站立场无关。仅供读者参考,并请自行核实相关内容。