コンテンツへスキップ

Tensorflow インストール ( tensorflow / tensorflow-gpu )

Tensorflowのインストール

Tensorflowは、Kerasのバックエンドに使われていたり、各種機械学習のフレームワークを利用する際に、よく出てきます。
個人的には、GPU対応パッケージが準備されており、そのサポート範囲が広いので重宝しております。

使用できるパッケージ

  • tensorflow - 最新の安定版リリース、CPU のみ(Ubuntu、Windows 用)
  • tensorflow-gpu - 最新の安定版リリース、GPU サポートを含む(Ubuntu、Windows 用)
  • tf-nightly - ナイトリー ビルドのプレビュー、CPU のみ(不安定)
  • tf-nightly-gpu - ナイトリー ビルドのプレビュー、GPU サポートを含む(不安定、Ubuntu と Windows 用)

Pip/Pip3コマンドでTensorFlow/TnesorFlow-GPUをインストール

インストールコマンドのサンプルを記載しておきます。

# Current stable release for CPU-only
pip3 install tensorflow

# Preview nightly build for CPU-only (unstable)
pip3 install tf-nightly

# Install TensorFlow 2.0 Beta
pip3 install tensorflow==2.0.0-beta0

以下が、Windows環境にGPU対応版をインストールした場合です。

@windows>pip3 install tensorflow-gpu
...
Successfully installed setuptools-41.0.1 tensorflow-gpu-1.14.0

参考にさせて頂いたサイト

pip での TensorFlow のインストール

コメントを残す

メールアドレスが公開されることはありません。 が付いている欄は必須項目です

CAPTCHA