禁用 GPU 使我的 CNTK 程序正常工作.为什么

disabling GPU makes my CNTK program work. Why?

本文关键字:工作 为什么 常工作 GPU 我的 CNTK 程序 禁用      更新时间:2023-10-16

我有一个使用 CNTK 的C++程序,可以在数十台电脑上正常工作。 但是在这台特定的PC上,当从磁盘加载神经网络时,它会崩溃 - 在程序启动时。 最后,我禁用 GPU(从窗口,而不是从任何 CNTK 代码(,然后它加载、运行并正常工作。 那么,为什么存在 GPU 会导致此问题 - 据我所知,我没有在我的 CNTK 代码中使用 GPU。有想法吗?谢谢!

回答我自己的问题:我检查了我的代码,然后注意到/回忆起 LoadModel 调用将 DeviceDescriptor 作为参数。 对于我的一些网络,我将其设置为DefaultDevice,即GPU。 我把它们改成了CPUDevice - 问题就消失了。 所以,一定是这个特定的GPU有问题——因为我的代码的旧版本在各种机器上都很好。 就我而言,我在评估时不需要 GPU,因此确保我只在运行时代码中指定 CPU 似乎更安全,以防万一其他系统出现像这台 PC 一样的"GPU 问题"(仍然不了解问题,真的(