Cntk无法通过cntk.all_devices()

时间:2017-02-09 09:19:53

标签: cntk

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。我怎么能解决这个问题?

1 个答案:

答案 0 :(得分:0)

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

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