環境
- windows10
- GPU(NDIVIA GeForce GTX 1060 3GB)
- Anaconda3
- CUDA
- CuDNN
手順
ドライバーの更新
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"
]