ai:keras
文書の過去の版を表示しています。
Keras
実行時に任意の backend を選択する方法。
USE_CNTK= 0 USE_TENSORFLOW= 1 USE_PLAIDML= 0 USE_CPU=0 if USE_CPU: os.environ['CUDA_DEVICE_ORDER']='PCI_BUS_ID' os.environ['CUDA_VISIBLE_DEVICES']='-1' if USE_PLAIDML: import plaidml.keras plaidml.keras.install_backend() elif USE_CNTK: os.environ['KERAS_BACKEND']='cntk' if USE_CPU: import cntk cpu= cntk.cpu() cntk.try_set_default_device( cpu ) elif USE_TENSORFLOW: from tensorflow import keras from tensorflow.keras import layers from tensorflow.keras import models from tensorflow.keras import optimizers from tensorflow.keras import backend as K if not USE_TENSORFLOW: import keras from keras import layers from keras import models from keras import optimizers from keras import backend as K
ai/keras.1561785806.txt.gz · 最終更新: 2019/06/29 14:23 by oga