import tensorflow as tfgpu_options=GPUOptions(per_process_gpu_memory_fraction=0.3)sess=tf.Session(config=tf.ConfigProto(gpu_tions=gpu_options))tf.keras.backend.set_session(sess)2跟3同一行, 4跟5同一行我這樣設定可以正常限制記憶體
https://goo.gl/Cn2uak 可以參考這個如果是0吃30% 1吃0%,一開始設定os.environ就好但如果是0吃30% 1吃10%這種,可能要再找找之前用keras.backend去設定記憶體時,也有遇過設定了os.environ,但也會像這樣占用兩個GPU改用tf.keras.backend則不會