2019-06-04から1日間の記事一覧

複数のgpu が接続されているときの環境変数の設定

Python3

nvidia-smi で複数のgpuが存在するとき import os os.environ["CUDA_VISIBLE_DEVICES"] = "1" from keras import backend as K K.tensorflow_backend._get_available_gpus() ['/device:GPU:0'] nvidia-smi