Custom nvidia-docker for Mxnet and Pytorch.
Contains cuda+cudnn environment and deeplearning framework.
You can use them straightly to train or test.
- Mxnet-cuda90
docker pull pistonyang/mxnet:base-cuda90
- Mxnet-cuda100
docker pull pistonyang/mxnet:base-cuda100
- Pytorch
docker pull pistonyang/pytorch:base-cuda100