导入以下代码控制GPU内存占用,爆出与tensorflow2.0版本不兼容问题。
只需将下面的代码:
import os
import tensorflow as tf
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.3
keras.backend.tensorflow_backend.set_session(tf.Session(config=config))
改为:
import os
import tensorflow as tf
os.environ["CUDA_VISIBLE_DEVICES"] = "2"`
config=tf.compat.v1.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.3`
tf.compat.v1.keras.backend.set_session(tf.compat.v1.Session(config=config))
`
即可。