ContextQMD
Back to Mxnet

Docker images for MXNET

docker/README.md

1.9.13.0 KB
Original Source
<!--- Licensed to the Apache Software Foundation (ASF) under one or more contributor license agreements. See the NOTICE file distributed with this work for additional information regarding copyright ownership. The ASF licenses this file to you under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. -->

Docker images for MXNET

How to use

First make sure docker is installed. The docker plugin nvidia-docker is required to run on Nvidia GPUs.

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

For example, the following command launches a container with the Python package installed. It will pull the docker images from docker hub if it does not exist locally.

bash
docker run -ti --rm mxnet/python

Then you can run MXNet in python, e.g.:

bash
# python -c 'import mxnet as mx; a = mx.nd.ones((2,3)); print((a*2).asnumpy())'
[[ 2.  2.  2.]
 [ 2.  2.  2.]]

If the host machine has at least one GPU installed and nvidia-docker is installed, namely nvidia-docker run --rm nvidia/cuda nvidia-smi runs successfully, then you can run a container with GPU supports

bash
nvidia-docker run -ti --rm mxnet/python:gpu

Now you can run the above example in GPU 0:

bash
# python -c 'import mxnet as mx; a = mx.nd.ones((2,3), mx.gpu(0)); print((a*2).asnumpy())'
[[ 2.  2.  2.]
 [ 2.  2.  2.]]

Hosted containers

All images are based on Ubuntu 14.04. The gpu tag is built with CUDA 8.0 and cuDNN 5.

Python

Hosted at https://hub.docker.com/r/mxnet/python/

Python versions: 2.7.12 and 3.5.2.

Available tags:

  • mxnet/python
  • mxnet/python:gpu

R

Hosted at https://hub.docker.com/r/mxnet/r-lang/

R version: 3.3.3

Available tags:

  • mxnet/r-lang
  • mxnet/r-lang:gpu

Julia

Hosted at https://hub.docker.com/r/mxnet/julia/

Julia version: 0.5.1

Available tags:

  • mxnet/julia
  • mxnet/julia:gpu

Scala

Hosted at https://hub.docker.com/r/mxnet/scala/

Scala version: 2.11.8

Available tags:

  • mxnet/scala

Perl

Hosted at https://hub.docker.com/r/mxnet/perl/

Perl version: 5.18.2

Available tags:

  • mxnet/perl
  • mxnet/perl:gpu

How to build

The following command build the default Python package

bash
./tool.sh build python cpu

Run ./tool.sh for more details. Use

Tips: The following commands stop all docker containers and delete all docker images.

bash
docker stop $(docker ps -a -q)
docker rm $(docker ps -a -q)
bash
docker rmi $(docker images -a -q)