トップページ -> インストール,設定,利用 -> Windows -> TensorFlow 2のインストール(Windows 上)
[サイトマップへ], [サイト内検索へ]

TensorFlow 2のインストール(Windows 上)

ユースケース:Windows で TensorFlow バージョン 2 を使いたい場合

サイト内の関連ページ:

参考Webページ:

先人に感謝.

このWebページに記載しているプログラムは https://github.com/tensorflow/tensorflow#download-and-setup をもとに作成


前準備

Python のインストール

TensorFlow を使うので,https://pypi.org/project/tensorflow-gpu/#filesで,必要な Python のバージョンを確認しておくこと. 2019/12 時点では,Python 2.7 か 3.5 か 3.6 か 3.7

以下,Windows で Python をインストール済みであるものとして説明を続ける.

NVIDIA グラフィックスボード・ドライバ,NVIDIA CUDA ツールキットのインストール

※ Windows でのインストール手順の詳細は,「別のページ」で説明している

次の Web ページには,pip を用いて TenforFlow をインストールするときの CUDA ツールキット,cuDNN SDK のバージョン指定がある. 2020年1月時点の情報では,次の通り

https://www.tensorflow.org/install/gpu#pip_package

参考Webページ:

NVIDIA cuDNN のインストール

Windows でのインストール手順の詳細は,別ページで説明している

インストールの要点:https://developer.nvidia.com/cudnn から cuDNN を入手し,C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1 などに展開(解凍)し,パスを通しておくこと


TensorFlow 2のインストール手順(Windows 上)

  1. Windows で,コマンドプロンプトを管理者として実行

    [image]
  2. パッケージのアンインストール操作

    ※ トラブルの可能性を減らすために,次の操作でアンインストールを行っておく.

    pip uninstall -y tensorflow tensorflow-gpu keras
    

    [image]
  3. pip を用いてインストール

    ※ 「pip install」は,Python パッケージをインストールするためのコマンド

    GPU 版 Tensorflow 2 をインストールする場合

    ※ 「py」は,Windows のPythonランチャーAnaconda 3 内の Python などを間違って使ってしまわないように,Pythonランチャーを使って Python を起動している.

    py -m pip install --upgrade pip
    pip install -U tensorflow-gpu 
    

    [image]

    CPU 版の Tensorflow をインストールする場合

    ※ 「py」は,Windows のPythonランチャーAnaconda 3 内の Python などを間違って使ってしまわないように,Pythonランチャーを使って Python を起動している.

    py -m pip install --upgrade pip
    pip install -U tensorflow 
    

    [image]
  4. TensorFlow のバージョン確認

    ※ バージョン番号が表示されれば OK.下の図とは違うバージョンが表示されることがある

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

    [image]
  5. (GPU を使うとき) GPU が認識できてるかの確認

    py -c "from tensorflow.python.client import device_lib; print(device_lib.list_local_devices())"
    

    次のように「device_type: "GPU"」があれば,GPUが認識できている.

    [image]

TensorFlow のプログラム例

  1. TensorFlow がインストールできたかを確認したい.

    Windows で Python プログラムを動かす.

    ※ そのために, Windows では,「python」コマンドやPythonランチャーである「py」を使う. あるいは, PyCharmなどにある Python コンソールも便利である.

    結果として 「b'Hello, TensorFlow!'」のように表示されるので確認する.

    import tensorflow as tf
    hello = tf.constant('Hello, TensorFlow!')
    print(hello)
    

    [image]

    ここでエラーメッセージが出ることがある. tensorflow GPU 版に合致する NVIDIA CUDA ツールキットがインストールされていないことが原因かも知れない.

    次の Web ページには,pip を用いて TenforFlow をインストールするときの CUDA ツールキット,cuDNN SDK のバージョン指定がある. 2020年1月時点の情報では,次の通り

    • CUDA ツールキットのバージョン: 10.1
      指定されているバージョンより高いものは使わないほうが良いでしょう.
    • cuDNN SDK のバージョン: 7.6 以上

    https://www.tensorflow.org/install/gpu#pip_package

    Windows でのCUDA ツールキットのインストールについては・「別のページ」で説明している.

  2. 今度は,次の Python プログラムを実行してみる.

    結果として 「42」のように表示されるので確認する.

    「tf.add(a, b)」と「a + b」は,同じ結果になる

    import tensorflow as tf
    a = tf.constant(10)
    b = tf.constant(32)
    print( tf.add(a, b) )
    print( a + b )
    

    [image]
  3. 今度は,次の Python プログラムを実行してみる.

    結果として 12 が表示されるので確認する.

    import tensorflow as tf
    matrix1 = tf.constant([[3., 3.]])
    matrix2 = tf.constant([[2.],[2.]])
    print( tf.matmul(matrix1, matrix2) )
    

    [image]

本サイトのサイトマップは,サイトマップのページをご覧下さい. 本サイト内の検索は,サイト内検索のページをご利用下さい.

問い合わせ先: 金子邦彦(かねこ くにひこ) [image]