keras运行时指定显卡及限制GPU用量

 

import os
import tensorflow as tf
import keras.backend.tensorflow_backend as KTF

#进行配置，每个GPU使用60%上限现存
os.environ["CUDA_VISIBLE_DEVICES"]="1,2" # 使用编号为1，2号的GPU
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.6 # 每个GPU现存上届控制在60%以内

#session = tf.Session(config=config)

# 设置session
#KTF.set_session(session )

 

https://blog.csdn.net/A632189007/article/details/77978058

https://blog.csdn.net/github_36326955/article/details/79910448