search
LoginSignup
36

More than 3 years have passed since last update.

posted at

windows10にTensorFlow GPUを入れる

環境

  • windows10
  • GPU(NDIVIA GeForce GTX 1060 3GB)
  • Anaconda3
  • CUDA
  • CuDNN

手順

ドライバーの更新

NDIVAの公式サイトよりドライバを更新

Anaconda3(仮想環境)

AnacondaのダウンロードページからAnaconda3をダウンロード
完了すると「Anaconda Navigator」があるので、ここで仮想環境を作れる。(コマンドでも作れます)

CUDA

CUDAのダウンロードページ
何も操作しない場合はこのディレクトリ
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2

CuDNN

CUDAのバージョンに合わせたCuDNNをダウンロードCuDNNのダウンロードページ
CuDNNはZipファイルでダウンロードできます。
zipファイルから展開したものをCUDA/v9.2ファルダに入れようとしましたが、パスで問題があったようなので展開したファイルをCUDAの該当するフォルダーにコピーしました。

ex) cuda/bin/cudnn64_7.dll → CUDA/bin/cudnn64_7.dll

pipでインストール

pip install tensorflow-gpu

Tensorflowのバージョンを確認して、ちゃんと入れられたかチェックします。

 python -c "import tensorflow as tf; print(tf.__version__)"

次に、GPUを認識できているかチェック

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

結果

2018-08-19 13:25:37.227703: I T:\src\github\tensorflow\tensorflow\core\platform\cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2
2018-08-19 13:25:37.534817: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:1405] Found device 0 with properties:
name: GeForce GTX 1060 3GB major: 6 minor: 1 memoryClockRate(GHz): 1.7085
pciBusID: 0000:01:00.0
totalMemory: 3.00GiB freeMemory: 2.42GiB
2018-08-19 13:25:37.539355: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:1484] Adding visible gpu devices: 0
2018-08-19 13:27:42.397650: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:965] Device interconnect StreamExecutor with strength 1 edge matrix:
2018-08-19 13:27:42.400826: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:971]      0
2018-08-19 13:27:42.402037: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:984] 0:   N
2018-08-19 13:27:42.403457: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:1097] Created TensorFlow device (/device:GPU:0 with 2123 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1060 3GB, pci bus id: 0000:01:00.0, compute capability: 6.1)
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 4112415611464959698
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 2226402099
locality {
  bus_id: 1
  links {
  }
}
incarnation: 11326046508655942687
physical_device_desc: "device: 0, name: GeForce GTX 1060 3GB, pci bus id: 0000:01:00.0, compute capability: 6.1"
]

参考にした記事

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
What you can do with signing up
36