dotnet / machinelearning

ML.NET is an open source and cross-platform machine learning framework for .NET.
https://dot.net/ml
MIT License
8.93k stars 1.86k forks source link

Add AMD/ROCM support to ML.NET #6899

Closed luisquintanilla closed 7 months ago

luisquintanilla commented 7 months ago

Issue originally reported in Model Builder repo by @Selbyl

dotnet/machinelearning-modelbuilder#2764

ROCM is being supported more on Windows recently.

I know AMD isn't as popular as Nvidia for AI, but is ther any future of updating ml.net to support PyTorch with ROCM?

luisquintanilla commented 7 months ago

In order for ML.NET to support this, TorchSharp and/or ONNX would have to support it. Currently, ONNX does through the Execution Providers.

However, they don't publish a NuGet similar to DirectML and OpenVINO.

In order to get this working in ML.NET and even then, this would only work for inference scenarios using ONNX Execution Providers, you'd need to:

Closing this issue for now.