Ubuntu20.04にNVIDIA Container Toolkitをインストールして、nvidia/cudaイメージを利用した簡単なDockerfileを作成して動作確認を行いました。docker上でGPUを利用するのが簡単になって嬉しいですね。

https://takake-blog.com/ubuntu-2004-install-nvidiacontainertoolkit/