1. 程式人生 > >如何在基於tensorflow的深度學習框架keras中指定GPU

如何在基於tensorflow的深度學習框架keras中指定GPU

import os

os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" 
os.environ["CUDA_VISIBLE_DEVICES"]="0,1"

使用0號, 1號GPU