CUDAとcuDNNとUbuntuのバージョンを指定してコンテナを立てる[nvidia-docker2]
docker run -it --runtime=nvidia nvidia/cuda:8.0-cudnn5-devel-ubuntu16.04
- –runtime=nvidia を忘れるとnvidia-smiが使えないので忘れないように
- cuDNN5.1などのイメージは用意されていない
- フレーバーはdevelにセットしておくのが無難 runtimeではnvccが使えないので注意[nvidia-docker2] – Vastee’s TIPS
- Nvidia-docker 提供イメージ一覧: https://hub.docker.com/r/nvidia/cuda/