keras GPU 使用
1、指定GPU import os os .environ[ "CUDA_VISIBLE_DEVICES" ] = "2" 2、按需分配 import tensorflow as tf import keras .backend .tensorflow _backend as KTF config = tf .ConfigProto () config .gpu _options .allow _growth=True #不全部占满显存, 按需分配 // config .gpu _options .per _process_gpu_memory_fraction = 0.3 #指定分配30%空间 session = tf .Session (config=config) # 设置session KTF .set _session(sess) 文章来源: keras GPU 使用