引入TorchSharp-cuda-windows后无法构建ML.NET GPU训练项目的问题求助
引入TorchSharp-cuda-windows后无法构建ML.NET GPU训练项目的问题求助
各位大佬好,我现在遇到一个棘手的问题:我想在项目里用ML.NET结合GPU来训练基于TorchSharp的NER模型,为此已经安装了适配的NVIDIA CUDA版本。
我运行了以下命令查看CUDA版本:
nvcc --version
输出结果如下:
Cuda compilation tools, release 12.1, V12.1.66
Build cuda_12.1.r12.1/compiler.32415258_0
目前我的项目引用了这三个NuGet包:
Microsoft.ML (3.0.1)
Microsoft.ML.TorchSharp (0.21.1)
Torchsharp-cuda-windows (0.102.2)
之前用Torchsharp-cpu (0.102.2)的时候,整个项目运行完全正常,但换成CUDA版本之后就没法正常构建了。而且没有明确的错误提示,只是偶尔构建时会出现这样的信息:
Child node "2" exited prematurely. Shutting down.
我试过相关的解决方案,但在我的场景里不管用。如果需要更多的环境信息或者日志,我随时可以提供,麻烦大家帮忙看看问题出在哪了!
备注:内容来源于stack exchange,提问作者JackFord
相关产品推荐
相关产品推荐

