『AWSのEC2インスタンスでdockerからGPUを使う方法』アシスタントとして回答する

AWSのEC2インスタンス上でDockerコンテナからGPUを利用するためには、以下の手順を順番に実施する必要があります。

1.適切なインスタンスの選択

まず、GPUを搭載したEC2インスタンスタイプを選択します。代表的なものには以下のようなものがあります:

• g4dn.xlarge: 16GiBのGPUメモリを持つ、NVIDIA T4 GPUを1つ搭載しています。
• p3.2xlarge: 16GiBのGPUメモリを持つ、NVIDIA V100 GPUを1つ搭載しています。

これらのインスタンスタイプは、GPUを必要とするワークロードに適しています。ただし、これらのインスタンスを利用するためには、AWSサポートへの申請が必要です。

2.OSの選択

インスタンスのOSとしては、 Ubuntu 22.04 LTS などの一般的なLinuxディストビューションが推奨されます。これらはNVIDIAドライバやDockerのインストールに適しています。

あなたの手慣れたLinuxディストリビューションを使うことをお勧めしますが、 ここではUbuntuを使っているものとします。

3.NVIDIAドライバのインストール

GPUを利用するためには、ドライバのインストールが必須です。以下の手順で進めましょう。

• 開発ツールのインストール

sudo apt update sudo apt install -y build-essential

• ドライバのダウンロード NVIDIAの公式サイトなどから、使用しているGPUに対応したドライバをダウンロードします。

• ドライバのインストール ダウンロードしたドライバを実行可能な状態にして、インストールします。

chmod +x NVIDIA-Linux-x86_64-*.run sudo ./NVIDIA-Linux-x86_64-*.run

インストール中のプロンプトには適宜対応してください。

4.Dockerのインストール

次にDockerをインストールします。

sudo apt update sudo apt install -y docker.io sudo systemctl start docker sudo systemctl enable docker

5.NVIDIA Container Toolkiのインストール

DockerコンテナからGPUを利用するためにNVIDIA Container Toolkitを導入しましょう。

GPGキーの追加→リボジトリの追加→パッケージの更新/インストール

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - distribution=$(. /etc/os-release;echo $ID$VERSION_ID) curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list sudo apt update sudo apt install -y nvidia-docker2 sudo systemctl restart docker

6.Dockerの起動

お疲れ様でした！インストールは以上です。 GPUを使用するDockerコンテナを起動する際には --gpusオプションを使用します。

nvidia-smiコマンドでGPUの状況を確認することができます。