nvidia/cuda
https://hub.docker.com/r/nvidia/cuda
CUDA and cuDNN image
CUDAが使えるイメージ
Pythonを入れて、pip installしていく必要がある
Ubuntuならaptで入れる（
python3 -m pip install
）
https://gitlab.com/nvidia/container-images/cuda