tree: 9897dc7e6c1f826f303126866621ca5e50c6f60d [path history] [tgz]
  1. cpu/
  2. cuda/
  3. README.md
docker/README.md

Docker images for MXNET

Pre-built docker images are available at https://hub.docker.com/r/dmlc/mxnet/

How to use

  1. First pull the pre-built image

    docker pull dmlc/mxnet
    
  2. Then we can run the python shell in the docker

    docker run -ti dmlc/mxnet python
    

    For example

    $ docker run -ti dmlc/mxnet python
    Python 2.7.6 (default, Jun 22 2015, 17:58:13)
    [GCC 4.8.2] on linux2
    Type "help", "copyright", "credits" or "license" for more information.
    >>> import mxnet as mx
    import mxnet as mx
    >>> quit()
    quit()
    

    Note: One may get the error message libdc1394 error: Failed to initialize libdc1394, which is due to opencv and can be ignored.

  3. Train a model on MNIST to check everything works

    docker run dmlc/mxnet python /mxnet/example/image-classification/train_mnist.py
    

If the host machine has Nvidia GPUs, we can use dmlc/mxnet:cuda, which has both CUDA and CUDNN installed. To launch the docker, we need to install nvidia-docker first.

  1. Pull the image

    docker pull dmlc/mxnet:cuda
    
  2. Train MNIST on GPU 0

    nvidia-docker run dmlc/mxnet:cuda python /mxnet/example/image-classification/train_mnist.py --gpus 0
    

How to build

docker build -t dmlc/mxnet:cpu cpu
docker build -t dmlc/mxnet:cuda cuda