tensorflow gpuをwindowsにインストールする方法
tensorflow(gpu版)をwindows 7に入れる方法についてのメモ。
環境は以下の通り。
1)OpenCVのインストール
(1-1) ここでopencv_python-3.4.1-cp36-cp36m-win amd64.wmlをダウンロード。
(1-2) ダウンロードしたファイルがある場所に移動し、以下のコマンドをうつ。
以下のサイトを参考にした。
OpenCV3.4とPython3.6をAnacondaでWndows10へインストール
2)tensorflow gpuのインストール
(2-1) ここから、win-64/tensorflow-gpu-1.8.0-py36_0.tar.bz2をダウンロードする。
(2-2) データのある場所に移動して、以下のコマンドでインストールする。
conda install tensorflow-gpu-1.8.0-py36_0.tar.bz2
(2-3) pythonを起動して、以下のコマンドでtensorflowをimportする。
import tensorflow as tf
エラーが出て、とあるバージョンのCUDAを入れて下さいと出るはず。
自分の場合は、tensorflow 1.8でCUDA 9.0を入れろと言われた。
3)CUDAのインストール
ここ(https://developer.nvidia.com/cuda-toolkit-archive)から、上記で確認したバージョンのCUDAをダウンロードし、インストールする。
自分の場合)
CUDA Toolkit 9.0
から
と進んで
Base Installer
をダウンロードし、インストールした。そうすると、
C:Program Files/NVIDIA GPU Computing Toolkit
ができているはず。続いて、
Patch 1, Patch 2, Patch 3
をダウンロードし、インストールした。
4)cuDNNのインストール
(4-1) 事前にNVIDIAのユーザー登録を済ましておく。
(4-2) ここ(https://developer.nvidia.com/rdp/cudnn-download)から、ダウンロードしたCUDAに対応するcuDNN Libraryをダウンロードし、解凍する。
自分の場合はDownload cuDNN v7.1.4 (May 16, 2018), for CUDA 9.0を選択した。
(4-3) 解凍するとcudaというフォルダーがデスクトップに現れるはず。
この内部のbinを3)でインストールしたCUDAの中のbinと一つにまとめる。
自分の場合は、cuda内にあるbinを
C:Program Files/NVIDIA GPU Computing Toolkit/CUDA/ver9.0/
に放り込んだ(はず。うろ覚え。)。
以下を参考にした。
5)その後の手順
後はCPU版と同様にprotobufとabslを入れる。以下の記事を参照のこと。