GPUによるコンテナのアクセラレーション」がサポートされました。Toolkitを使うことでを利用できるようになります。ここでは、それに必要な初期設定を行います。動作確認してみる手順0:前提条件の確認Dockerの「GPUによるコンテナのアクセラレーション」を使うにあたり、環境を確認しておきましょう。古すぎる環境ではGPUコンテナは使えません。詳細は以下のページをご覧ください。手順1:DockerのインストールまだDockerを入れてない場合は、以下の手順に従いインストールします。DriverのインストールGPUコンテナ(CUDAコンテナ)を実行するには、Nvidiaドライバーが必要です。Nvidiaドライバーのバージョンは、使用したいCUDAのバージョンに対応する以上のものにします(例えばtensorflowなどでは使うバージョンにより必要なCUDAバージョンが厳密に異なるため)。Toolkitを別途インストールしなくても大丈夫です。Driverのインストール方法は以下にまとめました。手順3:DockerでGPUコンテナを使えるようにする端末から以下入力します。nvidia-smi上手くいけばこんな表示がでます。–gpus」オプションをつけます。runの前にsudoを加えないと実行されないかもしれません。以上、UbuntuでDockerのGPU対応コンテナ利用に必要な初期設定でした。コメントをキャンセルメールアドレスが公開されることはありません。