GPU support

TensorFlow GPUサポートには、さまざまなドライバーとライブラリが必要です。 インストールを簡素化し、ライブラリの競合を避けるために、GPUサポート付きのTensorFlow Dockerイメージ(Linuxのみ)を使用することをお勧めします。 このセットアップでは、NVIDIA® GPU ドライバのみが必要です。

The install instructions are for the latest release of TensorFlow.このインストール手順は、TensorFlow の最新リリース用です。 古い TensorFlow リリースで使用する CUDA® および cuDNN バージョンについては、テスト済みのビルド構成を参照してください。

Pip パッケージ

利用できるパッケージ、システム要件、および手順については、pip インストール ガイドを参照してください。 TensorFlow pip パッケージには、CUDA® 対応カードの GPU サポートが含まれています。

pip install tensorflow

TensorFlow の最新の安定版リリースにおける GPU サポートとインストール手順を説明します。

pip install tensorflow==1.15 # CPUpip install tensorflow-gpu==1.15 # GPU

ハードウェア要件

以下の GPU 対応デバイスをサポートしています。

• NVIDIA® GPU カード、CUDA® アーキテクチャ 3.5, 5.0, 6.0, 7.0, 7.5, 8.0 および 8.0 より高いもの。 サポートされていない CUDA® アーキテクチャの GPU、PTX からの JIT コンパイルを回避する、または異なるバージョンの NVIDIA® ライブラリを使用する場合は、ソースからの Linux ビルド ガイドを参照してください。 (詳細はApplicationCompatibilityを参照)

Software requirements

The following NVIDIA® software must be installed on your system:

• NVIDIA® GPU drivers -CUDA®11.0 requires 450.x or higher.
• CUDA® Toolkit -TensorFlow supports CUDA®11 (TensorFlow >= 2.4.X) -TensorFlowはCUDA®11をサポートします(TensorFlow=2.4.X)。0）
• CUPTIにはCUDA®Toolkitが同梱されています。
• cuDNN SDK 8.0.4 cuDNNバージョン）
• （オプション）TensorRT 6.0 一部のモデルにおける推論のレイテンシおよびスループットを改善することができます。

Linux セットアップ

以下の手順は、Ubuntu に必要な NVIDIA ソフトウェアをインストールする最も簡単な方法です。 ただし、TensorFlow をソースから構築する場合は、上記の必要ソフトウェアを手動でインストールし、-devel TensorFlow Docker イメージをベースとして使用することを検討してください。

CUDA® Toolkit に付属する CUPTI をインストールします。

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64

Install CUDA with apt

Ubuntu16.04 および 18.04 に CUDA® 11 (TensorFlow >= 2.4.0) をインストールする方法について説明します。

Ubuntu 18.04 (CUDA 11.0)

# Add NVIDIA package repositories# Install NVIDIA driver# Reboot. Check that GPUs are visible using the command: nvidia-smi# Install development and runtime libraries (~4GB)# Install TensorRT. Requires that libcudnn8 is installed above.

Ubuntu 16.04 (CUDA 11.0)

# Add NVIDIA package repositories# Add HTTPS support for apt-key# Install NVIDIA driver# Issue with driver install requires creating /usr/lib/nvidia# Reboot. Check that GPUs are visible using the command: nvidia-smi# Install development and runtime libraries (~4GB)# Install TensorRT. Requires that libcudnn7 is installed above.

Windows設定

上記のハードウェア要件とソフトウェア要件を参照してください。 Windows 用の CUDA® インストール ガイドをお読みください。

インストールされている NVIDIA ソフトウェア パッケージが上記のバージョンと一致していることを確認します。 特に、TensorFlow は cuDNN64_8.dll ファイルがないとロードされません。

CUDA® 、CUPTI、および cuDNN のインストール ディレクトリを %PATH% 環境変数に追加してください。 たとえば、CUDA® ツールキットが C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0 に、cuDNN が C:\tools\cuda にインストールされている場合、%PATH% を更新して一致させます。