CNTK 使用 python API 停用 gpu 锁定

Question

如何从 python API (2.0 beta 12.0) 中通过 cntk 停用 GPU 锁定，以便多个用户可以使用同一个 GPU？

如果我从 https://github.com/Microsoft/CNTK/tree/master/Tutorials 执行 cntk 教程 201B，训练工作正常，但所有其他用户/进程收到错误消息："Failed to lock GPU 0 for exclusive use" 在执行基于 cntk 的优化时。

Answer 1

您可以使用 GPU id 显式设置设备：

cntk.device.set_default_device(cntk.device.gpu(0))

这样您就可以在同一个 GPU 上 运行 多个作业。