ai:keras
文書の過去の版を表示しています。
Keras
実行時に任意の backend を選択する方法。
import os 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
USE_CNTK | USE_TENSORFLOW | USE_PLAIDML | USE_CPU | |
---|---|---|---|---|
0 | 0 | 0 | 0 | 通常版 keras 、~/.keras/keras.json の設定に従う |
0 | 1 | 0 | 0 | TensorFlow 組み込み keras + GPU |
0 | 1 | 0 | 1 | TensorFlow 組み込み Keras + CPU |
1 | 0 | 0 | 0 | 通常版 keras + CNTK + GPU |
1 | 0 | 0 | 1 | 通常版 Keras + CNTK + CPU |
0 | 0 | 1 | N/A | 通常版 Keras + Plaidml、CPU/GPU は plaidml-setup (~/.plaidml) の設定に従う |
ai/keras.1561786252.txt.gz · 最終更新: 2019/06/29 14:30 by oga