基本は以下の記事に書いてあるDockerfileで大丈夫だが、このままだとGPUが使えないという問題がある。
GPUを使えるようにするにはベースイメージのところを
FROM nvidia/cuda:9.0-base-centos7
に変更する。
このとき、cuda9.0のbase imageを使うのがポイントで9.1以降だとGPUが使えない。
nvidia/cudaのDockerHubではUbuntuとCentOSのイメージが用意されているのでどちらかのOSを使っている場合はこれらをbase imageにするといい。