Jonathan Calmels edited this page Nov 16, 2017 · 14 revisions

Description

Docker image based on DIGITS.

Requirements

See the requirements for CUDA since the DIGITS image is based on a cuDNN runtime image.

Examples

# Run DIGITS on host port 5000
docker run --runtime=nvidia --name digits -d -p 5000:5000 nvidia/digits

If you want to use a dataset stored in a host directory, you will need to import it inside the container using a volume:

# Run DIGITS with the mnist dataset stored on the host under /opt/mnist
docker run --runtime=nvidia --name digits -d -p 5000:5000 -v /opt/mnist:/data/mnist nvidia/digits

If you want to persist jobs across multiple DIGITS containers, you can use a named volume. For DIGITS 3.0:

# Run DIGITS storing jobs in a host volume named digits-jobs

# For DIGITS 3.0
docker run --runtime=nvidia --name digits -d -p 5000:34448 -v digits-jobs:/usr/share/digits/digits/jobs nvidia/digits:3.0

# For DIGITS 3.3 and 4.0
docker run --runtime=nvidia --name digits -d -p 5000:34448 -v digits-jobs:/jobs nvidia/digits:4.0

# For DIGITS 5.0
docker run --runtime=nvidia --name digits -d -p 5000:5000 -v digits-jobs:/jobs nvidia/digits:5.0

# For DIGITS 6.0
docker run --runtime=nvidia --name digits -d -p 5000:5000 -p 6006:6006 -v digits-jobs:/jobs nvidia/digits:6.0

Tags available

Check the DockerHub

You can’t perform that action at this time.
You signed in with another tab or window. Reload to refresh your session. You signed out in another tab or window. Reload to refresh your session.
Press h to open a hovercard with more details.