Skip to content

Latest commit

 

History

History
20 lines (17 loc) · 431 Bytes

File metadata and controls

20 lines (17 loc) · 431 Bytes

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