关于tensorflow训练GPU占用问题

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/xiaomifanhxx/article/details/81538125

大家在训练数据的时候，经常会遇到tensorflow会占用发现的所有GPU的情况，这样会影响其他人使用GPU，因此我们可以在使用时设定选用的GPU。

import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"   # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"]="0"

这样我们就可以避免tensorflow占用所有GPU的情况。