ユーザ用ツール

サイト用ツール


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 0 0 1 独立版 keras + CPU 、~/.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) の設定に従う
  • keras.json の backend が “tensorflow” なら、全部 0 のときは 「独立版 keras + tensorflow」になります。
ai/keras.txt · 最終更新: 2019/06/29 14:34 by oga