问题:在keras里插入一下代码,限制一下GPU节点使用和占用内存量,发现根本不起作用
import tensorflow as tf
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
from keras.backend.tensorflow_backend import set_session
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.5
set_session(tf.Session(config=config))
解决办法:玄学,卸载keras重新安装就好了