深度学习
2020-12-03
1. 避免tensorflow占用全部GPU
使用tensorflow时，启用GPU支持会占用所有的GPU，虽然每个GPU都只占用了一点，GPU占用情况可以用如下命令查看：
nvidia-msi
解决GPU全占的办法是加入以下代码
os.environ['CUDA_DEVICES_ORDER'] = 'PCI_BUS_ID'
os.environ['CUDA_VISIBLE_DEVICES'] = '0' # 使用 0 GPU
config = tf.ConfigProto()
#config.gpu_options.per_process_gpu_memory_fraction = 0.9 # 控制使用每个GPU的大小
config.gpu_options.allow_growth = True