NVIDIA公式CUDAイメージに任意のPythonをインストールするDockerfile

DockerGPU周りの実験をしててタイトルのやつが欲しくなったので作った。

1個目がタイトル通りのやつ。なるべく環境は汚さないようにPythonは/python/$PYTHON_VERSION以下にインストールしてる。
2個目が備忘録的なおまけでmultistage buildにしてその中で作業できるようにしたやつ。今回はpoetryを使った例をのっけてある。
使うときはdocker build --target main -t name:tag .みたいにすれば良いと思う。
gist.github.com