在使用GPU版的TensorFlow跑程序的时候，如果不特殊写代码注明，程序默认是占用所有主机上的GPU，但计算过程中只会用其中一块。也就是你看着所有GPU都被占用了，以为是在GPU并行计算，但实际上只有其中一块在运行；另外的所有显卡都闲着，但其显存都被占用了，所以别人也用不了。不过这种情况通过在程序之前加三行代码就可以解决：在使用GPU版的TensorFlow跑程序的时候，如果不特殊写代码注明，程序默认是占用
阅读背景：
关于TensorFlow的GPU设置
来源：互联网
分享到：
