AWSでtensorflowがGPUを認識しないと思ったら

  • 1
    いいね
  • 0
    コメント

AWSでディープラーニング用のEC2インスタンスを用意して、テストコードを動かしたら、GPUを認識してくれなかった。

いろいろ調べた結果、CPU用のtensorflowを使っていたことに気づいた・・・。

$ conda install tensorflow-gpu

で解決(anacondaの場合)。pipからインストールするにはこちらを参照。