トップページ -> コンピュータ -> Ubuntu -> TenforFlow GPU 版のインストール(Ubuntu 18.04 上)
[サイトマップへ], サイト内検索:

TenforFlow GPU 版のインストール(Ubuntu 18.04 上)

サイト構成 連絡先,業績など 実践知識 コンピュータ 教材 サポートページ

次のページの記載に従いインストールする.

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

先人に感謝.


前準備

Ubuntu のシステム更新

端末で,次のコマンドを実行.

sudo apt update 
sudo apt -yV upgrade
sudo shutdown -r now

Python3 開発用ファイル,セットアップツール,pip,numpy のインストール

端末で,次のコマンドを実行.

sudo apt -yV install python3-dev python3-setuptools python3-pip python3-numpy

NVIDIA グラフィックスカードの確認とドライバのインストール,NVIDIA CUDA ツールキット,NVIDIA cuDNN 再配布パッケージのインストール

Ubuntu での手順は,「別のページ」で説明している. NVIDIA グラフィックスカードがある場合のみ行うこと.


TensorFlow GPU 版のインストール

  1. インストール

    端末で,次のコマンドを実行.

    sudo pip3 install tensorflow-gpu 
    

  2. その結果,エラーメッセージが出ていないことを確認.

  3. TensorFlow のバージョン確認

    端末で,次のコマンドを実行.

    ※ バージョン番号が表示されれば OK.

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

    TenforFlow GPU 版がうまくインストールできない場合

    次の Web ページには,pip を用いて TenforFlow をインストールするときの CUDA ツールキット,cuDNN SDK のバージョン指定の情報がある. 例えば,tenforflow 1.13 では・ CUDA ツールキットのバージョンは 10.0(10.1 は動かない),cuDNN SDK のバージョンは 7.4.1 以上の情報がある.

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

  4. keras のバージョン確認

    ※ バージョン番号が表示されれば OK.

    python3 -c "import keras; print( keras.__version__ )"
    

  5. TensorFlow の動作確認

    https://github.com/tensorflow/tensorflow に記載のサンプルプログラムを実行してみる

    Python プログラムを動かしたい. そのために,「Python コンソール」を使う.

    PyCharmspyder を使うのが簡単.

    python3
    

    Python プログラムを動かしたい. そのために,「Python コンソール」を使う.

    PyCharmspyder を使うのが簡単.

    エラーメッセージが出ていないことを確認.

    import tensorflow as tf
    tf.enable_eager_execution()
    tf.add(1, 2).numpy()
    hello = tf.constant('Hello, TensorFlow!')
    hello.numpy()