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