tensorflow gpuをwindowsにインストールする方法

tensorflow(gpu版)をwindows 7に入れる方法についてのメモ。

 

環境は以下の通り。

Windows 7 64 bit

gpu: NVIDIA Quadro K2200

python 3.6 (Anacondaを用いてインストールした)

tensorflow 1.8

CUDA 9.0

cuDNN v7.1.4

 

 1)OpenCVのインストール

(1-1) ここopencv_python-3.4.1-cp36-cp36m-win amd64.wmlをダウンロード。

 

(1-2) ダウンロードしたファイルがある場所に移動し、以下のコマンドをうつ。

pip install opencv_python-3.4.1-cp36-cp36m-win amd64.wml

 

以下のサイトを参考にした。

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

から

Operating System : Windows 

Architecture : x86_64

Version : 7

Install Type : exe(local)

と進んで

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というフォルダーがデスクトップに現れるはず。

この内部のbin3)でインストールしたCUDAの中のbinと一つにまとめる。

自分の場合は、cuda内にあるbinを

C:Program Files/NVIDIA GPU Computing Toolkit/CUDA/ver9.0/

に放り込んだ(はず。うろ覚え。)。

 

以下を参考にした。

Effectiveさお

 

5)その後の手順

後はCPU版と同様にprotobufabslを入れる。以下の記事を参照のこと。

haitenaipants.hatenablog.com