Ubuntu18.04にTensorFlowGPUをインストールする方法

はじめに

TensorFlow はオープンソースの機械学習プラットフォームです。 Googleは、研究者、データサイエンティスト、開発者が機械学習モデルを実装するプロセスを簡素化できるようにソフトウェアを設計しました。

この数値計算用のエンドツーエンドライブラリは、複数のCPU、GPU、およびモバイルオペレーティングシステムで実行できます。 このチュートリアルでは、Ubuntu18.04にTensorFlowをインストールする方法を学びます。

前提条件

• Ubuntu Linuxシステム（16.04以降）
• コマンドライン/ターミナルウィンドウへのアクセス（ Ctrl + Alt + T ）
• sudoのユーザーアカウント 特権
• Pip19.0以降

ステップ1：必要なパッケージをインストールする

TensorFlowをインストールする前に、Python開発環境をセットアップする必要があります。次のソフトウェアが含まれています：

• Python （バージョン3.4以降）
• pipパッケージマネージャー（バージョン19.0より前）
• Virtualenv （Python環境を分離するためのソフトウェア）

ターミナルウィンドウを開き、次のコマンドでリポジトリを更新することから始めます。

sudo apt update

次に、次のコマンドを使用して最初の2つのパッケージをインストールします。

sudo apt install python3-dev python3-pip

次に、Virtualenvをシステム全体にインストールするために、次のコマンドを実行します。

sudo pip3 install -U virtualenv

ステップ2：仮想環境の作成

UbuntuシステムにVirtualenvがインストールされたので、これを使用してPython環境を作成および分離できます。

新しい./venvで最初の環境を作成します ディレクトリ：

virtualenv --system-site-packages -p python3 ./venv

次に、仮想環境をアクティブ化して、仮想環境内での作業を開始します。次のコマンドを実行します：

source ./venv/bin/activate

これで、シェルプロンプトに（venv）が表示されます。 下の画像のようなプレフィックス：

venvをアクティブにしたら 、 pipのインストールに進みます 隔離された環境内：

pip install --upgrade pip

仮想環境内のすべてのパッケージの完全なリストを表示するには、次のコマンドを使用します。

pip list

次の画像のように、すべてのパッケージとそれぞれのバージョンが表示されます。

ステップ3：TensorFlowのインストール

次の手順は、 TensorFlow for CPUをインストールするかどうかによって異なります。 またはTensorFlowfor GPU 。選択は、ワークロードの性質とハードウェアオプションによって異なります。

オプション1：CPU用のTensorFlowをインストールする

デフォルトのTensorFlow ソフトウェアパッケージは、CPUベースのワークロードをサポートします。パッケージとその依存関係をインストールするには、次のコマンドを入力します。

pip install --upgrade tensorflow

オプション2：GPU用のTensorFlowをインストールする

GPU用TensorFlow 専用のNVIDIACUDA®対応GPUおよび関連するドライバーが必要です。このソフトウェアパッケージは、GPUベースの機械学習ワークロードを対象としています。

最新の安定バージョンのTensorFlowをインストールするには GPU用 、コマンドを実行します：

pip install --upgrade tensorflow-gpu

TensorFlow for GPUのハードウェアおよびソフトウェア要件のリストについては、GPUサポートに関するTensorFlowのドキュメントを参照してください。

オプション3：古いバージョンのTensorFlowをインストールする

古いバージョンのCPUおよびGPU用のTensorFlow ダウンロードも可能です。

バージョン1.14以前 次の形式でコマンドを実行することでインストールされます：

pip install tensorflow==package_version

TensorFlow forCPU1.14をインストールするには 、コマンドを実行します：

pip install tensorflow==1.14

TensorFlow forGPU1.14をインストールするには 、コマンドを実行します：

pip install tensorflow-gpu==1.14

バージョン1.15 CPUベースとGPUベースの両方のワークロードをサポートします。 TensorFlow 1.15をインストールするには 、次のコマンドを入力します：

pip install tensorflow-gpu==1.15rc2

ステップ4：TensorFlowのインストールを確認する

UbuntuにTensorFlowがインストールされていることを確認するには、ターミナルウィンドウで次のコマンドを入力します。

python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"