tensorflow默认训练时占用一块GPU的全部内存,在训练代码python文件中加入以下代码,可以避免这种问题,给实验室的小伙伴们留一点空间。
gpu_options = tf.GPUOptions(allow_growth=True)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
tensorflow默认训练时占用一块GPU的全部内存,在训练代码python文件中加入以下代码,可以避免这种问题,给实验室的小伙伴们留一点空间。
gpu_options = tf.GPUOptions(allow_growth=True)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))