過去にAWSの公式手順を参照する以下の記事を書いたのですが、nvidia-dockerがv2になって進化していたので更新します。 ECSでGPUを扱うためのAMI作成 by Amazon Linux Deep Learning AMI なぜ公式の方法ではなくnvidia-dockerを使いたいのか それにはまず以下の記事でnvidia-dockerとは何をしているのかを知りましょう。 nvidia-dockerは何をしているのか dockerからGPU扱う時には、volumeを指定してマウントしたり、deviceオプションで適切なdeviceを渡してあげるか、privilegedでホスト上のプロセスとほとんど同じレベルでホストへのアクセスを許可するか(AWS公式に記載されている方法)をする必要があります。 上記記事の通り、nvidia-dockerはGPUの情報を抽象化してdockerに渡

