Home > Net >  Why won't ML.NET Model Builder GPU extension install properly?
Why won't ML.NET Model Builder GPU extension install properly?

Time:10-29

When I use ML.NET Model Builder in VS2022 and try to configure the environment, the compatibility check informs me that the GPU extension is missing.

enter image description here

When I click "Install extension" it leads me to Visual studio's marketplace and suggests I download "ML.NET Model Builder GPU Support 2019" (not 2022). When I install the ML.NET Model Builder GPU Support 2022 manually the compatibility check still informs me that the GPU extension is missing. Is this a bug or am I missing something?

CUDA 10.1 and cuDNN 7.6.4 are both installed.

CodePudding user response:

GPU extension was not the issue as VS22 kept stating. Installing the ML.NET Model Builder 2022 (link) manually instead of through VS22 solved this problem for me.

CodePudding user response:

The Model Builder and GPU extension versions need to match. Installing the latest version of Model Builder extension and Model Builder GPU extension should resolve this.

Unfortunately in older version of Model Builder the error message doesn't state this.

  • Related