分类目录 学习
关于如何在机器上跑多GPU程序的解析
问题 在服务器里使用显卡跑程序的时候，经常有新手甚至老手也不太清除为什么程序跑起来了，但是感觉跑的很慢，调用了GPU好像没有调用成功一样，明明程序成功运行了。 如上图，PID 为28329进程运行程序时使用的 GPU 默认占满所有可用的显卡 GPU 和 G_MEM，但只在第一个 GPU 上进行计算。虽然调用了八块显卡资源，并且占用了所有 GPU 剩下的显存，但却只有一块真正在工作，大大浪费了资源，利用率低下。 解决方法很简单，TensorFlow 有一些参数可以简单控制显卡调用数量，可以通过在运 …