LoginSignup
10
12

More than 3 years have passed since last update.

keras、tensorflowをCPU版からGPU版に変更する

Last updated at Posted at 2019-01-03

発端

kerasの学習が妙に遅い、mnistすら遅いので確認した所、tensorflowがCPU版になっていました
pip installを色々する内にやってしまったのでしょう

原因

GPU版を入れている環境にCPU版を入れてしまったことが原因のようです
GPU版が同居するとCPU版が使われるっぽいですが、詳しくは見ていません
この原因を調べる事に価値は無さそうなので、直接解決します

解決

再インストールです
cpuだけuninstallしてもうまくいきません、tensorflowが無いと言われます
gpuも再インストールします

pip uninstall tensorflow
pip uninstall tensorflow-gpu
pip install tensorflow-gpu

テストコード

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

結果(実際は20行ぐらいあります)

2019-01-03 09:26:06.702413: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1432] Found device 0 with properties:
name: GeForce GTX 1070 major: 6 minor: 1 memoryClockRate(GHz): 1.7085

かくにん!よかった

補足

cpuになっている時の結果。

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 665334277417354455
]
10
12
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
10
12