当我在VS2022中使用ML.NET Model Builder并尝试配置环境时，兼容性检查通知我缺少GPU扩展。
当我点击“安装扩展”时，它会引导我进入Visual Studio的市场，并建议我下载“ML.NET Model Builder GPU Support 2019”（而不是2022）。当我手动安装ML.NET Model Builder GPU Support 2022时，兼容性检查仍然通知我缺少GPU扩展。这是一个bug还是我遗漏了什么？
CUDA 10.1和cuDNN 7.6.4均已安装。
2条答案
按热度按时间kxeu7u2r1#
模型构建器和GPU扩展版本需要匹配。安装最新版本的模型构建器扩展和模型构建器GPU扩展应该可以解决此问题。
不幸的是，在旧版本的模型生成器的错误消息没有说明这一点。
mfuanj7w2#
GPU扩展并不像VS22一直说的那样是个问题。手动安装ML.NET Model Builder 2022 (link)而不是通过VS22为我解决了这个问题。