Cntk 无法通过 cntk.all_devices() 检测到我的 gpu

Question

Cntk 通过调用 cntk.all_devices() 仅检测到 1 个设备（我的 cpu）。但是我的电脑上确实有一个GPU。通过运行cntk支持的教程，我可以得到一些信息：

-------------------------------------------------------------------
-------------------------------------------------------------------
GPU info:

                Device[0]: cores = 48; computeCapability = 2.1; type = "NVS 310"; memory = 512 MB
-------------------------------------------------------------------

##############################################################################
#                                                                            #
# Train command (train action)                                               #
#                                                                            #
##############################################################################


Model has 9 nodes. Using CPU.

因此，我无法通过调用 set_default_device(gpu(0)) 使用我的 gpu。我该如何解决这个问题？

Answer 1

CNTK 的最低 GPU 计算能力为 3.0。 （编辑： 您可以使用 cntk.exe 运行 教程这一事实表明 v1 可执行文件中某处存在错误。 ).当您 运行 使用 cntk.exe 教程时，它会打印出 GPU 信息，但最终仍会使用 CPU：Model has 9 nodes. Using CPU.

解决这个问题的唯一方法是改变常量MininumCCMajorForGpu in BestGpu.cpp的值并重新编译。