Skip To Content

GPU を使用するための ArcGIS Notebook Server の構成

グラフィカル プロセシング ユニット (GPU) を使用すると、機械学習モデルの処理時間をすばやく向上することができます。 ArcGIS Notebook Server で追加の手順を実行すると、ホスト コンピューターで NVIDIA GPU を活用できるようになります。

注意:

10.8 以降、組み込み ArcGIS Notebook Server ランタイムは、GPU サポートを有効化するための Conda CUDA Toolkit を含むようになりました。 以前は、このワークフローは、CUDA を含むために、カスタム ランタイムを構築する必要がありました。

以下のワークフローには、主に 2 つの目的があります。 1 つは、サイトの Docker コンポーネントが GPU 対応のコンテナーを構築できるようにする NVIDIA ドライバーおよびランタイムをインストールすることです。 もう 1 つは、NVIDIA ランタイムを使用するように構成されたノートブック ランタイムのコピーを作成することです。 このランタイム使用して開かれたすべての ArcGIS Notebooks は、GPU 対応のコンテナー内で起動されます。 これに加えて、新しいランタイムは、ノートブック ランタイムの Python ライブラリをすべて維持します。

ArcGIS Notebook Server のインストールと構成を完了したら、次の手順を実行します。 ArcGIS Notebook Server サイトに複数のコンピューターが含まれている場合、すべてのコンピューター上でステップ 1 ~ 3 を実行します。

  1. 適切な NVIDIA ドライバーを、サイトの各コンピューターにインストールします。 詳細については、NVIDIA Web サイトをご参照ください
  2. ノートブックのコンテナーが GPU を活用できるよう、コンピューター上に nvidia-docker 2.0 ランタイムをインストールします。 OS 別のダウンロードとドキュメントについては、GitHubNVIDIA-Docker リポジトリをご参照ください。
  3. 各コンピューター上で次のコマンドを実行し、NVIDIA の要素が適切にインストールされていることを確認します。

    docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi

  4. 管理者として ArcGIS Enterprise ポータルにサイン インし、ArcGIS Notebook Server Manager を開きます
  5. [設定] ページを開き、[ランタイム] をクリックします。
  6. GPU 対応ランタイムのベースとして使用するランタイム ([ArcGIS Notebook Python 3 Advanced] または [ArcGIS Notebook Python 3 Standard]) の [編集] ボタンをクリックします。 [画像 ID] 値をコピーします。 [キャンセル] をクリックして、エディターを終了します。
  7. [ランタイム] ページから、[ランタイムの登録] をクリックします。
  8. [ランタイムの登録] ページで、適切な名前 (「GPU Runtime」など) を入力し、バージョンを 11.1 と指定します。 [画像 ID] の値に、ステップ 5 でコピーした値を追加します。
  9. [Docker ランタイム] の値を nvidia に設定します。 [ランタイムの登録] をクリックして確認します。
  10. ArcGIS Notebook ServerNVIDIA GPU を使用するよう正常に構成したことを確認します。 [ノートブックの作成と編集] 権限または [高度なノートブック] 権限を持つポータル メンバーとして、ステップ 6 で高度なランタイムを選択した場合は、空のノートブックを作成します。 ノートブックのランタイムを選択するときに、新しい GPU 対応のランタイムを選択します。 次の内容をノートブックのセルにコピーし、セルを実行します。

    import torch
    torch.cuda.is_available()
    torch.cuda パッケージを実行するには GPU が必要なので、出力は True として返されます。

  11. 新しいセルで次のコマンドを実行し、コンピューターの GPU 構成を確認します。

    !nvidia-smi

GPU を使用するためにサイトの容量を削除する場合は、ArcGIS Notebook Server Manager の設定タブでランタイム セクションに移動し、このワークフローで作成したランタイムを削除します。