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。我怎么能解决这个问题?
答案 0 :(得分:0)
CNTK的最低GPU计算能力为3.0。 (编辑: 您可以使用cntk.exe运行教程这一事实表明v1可执行文件中存在某个错误。)。当您使用cntk.exe运行教程时,它会打印出GPU信息,但最终仍会使用CPU:Model has 9 nodes. Using CPU.
解决此问题的唯一方法是更改常量MininumCCMajorForGpu in BestGpu.cpp的值并重新编译。