dotnet / dotnet/machinelearning
How can I tell if ML.NET is using GPU acceleration?
Open
area-AutoML
- Dominant language
- C#
- Stars
- 9.4k
- Forks
- 2k
- Avg merge
- 2d 20h
- Merged PRs (30d)
- 11
Description
Hey Guys,
I have a question about GPU support in ML.NET.
My environment is:
• Visual Studio 2022 (Preview)
• .NET 6.0 (Preview)
I have downloaded and installed “ML.NET Model Builder GPU Support” and tested the GPU with Model Builder. Model Builder likes my GPU card as you can see:
It works o.k.
I have some legacy ML.NET code, and I am wondering if I re-build my old code will it use the newly installed GPU drivers?
How can I tell if my ML.NET project uses the GPU?
Charles
Contributor guide
Assessment
This issue has not been assessed yet.