WinML or DirectML support planned?

microsoft / CNTK

Microsoft Cognitive Toolkit (CNTK), an open source deep-learning toolkit

https://docs.microsoft.com/cognitive-toolkit/

Other

17.5k stars 4.29k forks source link

WinML or DirectML support planned? #3103

Open oscarbg opened 6 years ago

oscarbg commented 6 years ago

Windows only, but this should support AMD and Intel GPUs using D3D12 backend or even DirectML which with custom metacommands seems even able to use tensor cores on Volta GPUs.. Hope to hear news at least by Build conference in May

veikkoeeva commented 6 years ago

Cross-referencing https://github.com/Microsoft/CNTK/issues/960 and https://github.com/Microsoft/CNTK/issues/2839 as relevant.

oscarbg commented 6 years ago

@veikkoeeva note Net & C# are really not needed WinML is a Win32 API in Windows insider SDKs and this month 1803 SDK it has C++ header WinML.h and WinML.lib.. altough still no samples using Win32 API presumably at Build some Win32 API samples released.. See how Win32 API: 847d82e6-609c-4f88-9c94-0b4985a17582 Taken from (with DirectML details): https://blogs.msdn.microsoft.com/directx/2018/03/19/gaming-with-windows-ml/

ke1337 commented 6 years ago

WinML uses DirectML, and it's mainly for inference. Currently you can use WinML by exporting the CNTK model in ONNX format.

veikkoeeva commented 6 years ago

It appears ONNX and WinML are in https://github.com/dotnet/machinelearning/.