Power Systems サーバー上の Linux 用の NVIDIA GPU パッケージのインストール

このタスクを使用して、Power Systems™ サーバー上の Linux 用の NVIDIA GPU パッケージをインストールします。GPU 処理を可能にするには、必須の NVIDIA GPU パッケージをインストールする必要があります。

このタスクについて

Power8 システムの場合、次のタスクで説明されているとおりに CUDA 8 を使用します。Power9 システムの場合、次のタスクの CUDA 8 を CUDA 10 に置き換えます。

手順

  1. ご使用の NVIDIA GPU 用のドライバーをダウンロードしてインストールします。Power Systems サーバー上の Linux 用の NVIDIA ドライバー・リストは、NVIDIA Driver Downloads で確認できます。ダウンロード・ページのインストール手順に従ってください。
  2. 次のコマンドを使用して、CUDA リポジトリー・ファイルをダウンロードし、CUDA 8 をインストールします。
    wget https://developer.download.nvidia.com/compute/cuda/repos/rhel7/ppc64le/cuda-repo-rhel7-8.0.61-1.ppc64le.rpm
    rpm -i cuda-repo-rhel7-8.0.61-1.ppc64le.rpm
    yum clean all
    yum install cuda
  3. CUDA 8 の場合、NVIDIA CUDA Deep Neural Network ライブラリー cudnn-8.0-linux-ppc64le-v6.0-tgz を URL (https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v6/prod/8.0_20170307/cudnn-8.0-linux-ppc64le-v6.0-tgz) からダウンロードします。CUDA 10 の場合、NVIDIA CUDA Deep Neural Network ライブラリー cudnn-10.0-linux-ppc64le-v7.3.1.20.tgz を URL (https://developer.download.nvidia.com/compute/cuda/repos/rhel7/ppc64le/ cuda - 10.0.130 - 1.ppc64le.rpm) からダウンロードします。ファイルをダウンロードする前に、アカウントの作成とログインが必要な場合があります。
  4. 次のコマンドを使用して、cudnn-8.0-linux-ppc64le-v6.0.tgz ファイルを CUDA インストール・ディレクトリーに解凍します。
    sudo tar -xvf cudnn-8.0-linux-ppc64le-v6.0.tgz -C /usr/local
  5. 次のコマンドを使用して、環境変数を設定します。
    export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
    また、このコマンドを ~/.bashrc スクリプトに追加します。
  6. 次のコマンドを使用して、NVIDIA NCCL パッケージをインストールします。
    git clone https://github.com/NVIDIA/nccl.git
    cd nccl
    sudo make install -j4