ASRT 双GPU训练这样设置

SpeechModel251_p.py

 

if(__name__=='__main__'):
    
    import tensorflow as tf
    from keras.backend.tensorflow_backend import set_session
    #os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"
    #进行配置，使用70%的GPU
    config = tf.ConfigProto()
    #config.gpu_options.per_process_gpu_memory_fraction = 0.95
    #config.gpu_options.allow_growth=True   #不全部占满显存, 按需分配
    #set_session(tf.Session(config=config))
    
    
    datapath =  abspath + ''
    modelpath =  abspath + 'model_speech'