keras 如何设置多GPU训练

设置单卡训练。
直接在终端：

CUDA_VISIBLE_DEVICES=1 python train.py

或者在程序开头：

import os

os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"   

os.environ["CUDA_VISIBLE_DEVICES"]="0"#单卡
os.environ["CUDA_VISIBLE_DEVICES"]="0，1，2"#多卡