deeplearning-nvidia-docker(with autobuilding by docker hub)

Custom nvidia-docker for Mxnet and Pytorch.

Contains cuda+cudnn environment and deeplearning framework.

You can use them straightly to train or test.

Usage

• Mxnet-cuda90

docker pull pistonyang/mxnet:base-cuda90

• Mxnet-cuda100

docker pull pistonyang/mxnet:base-cuda100

• Pytorch

docker pull pistonyang/pytorch:base-cuda100