iterait/tensorflow repository overview

⁠dockerfiles

This is a repository contaning the open-source docker images provided by Iterait a.s.

We use nvidia-docker instead of regular docker in order to enhance GPU support. However, regular docker might be used as well.

⁠DockerHub

All built images are stored in Iterait DockerHub⁠.

⁠Build

All images may be built by regular docker build process.

$ docker build -t <image-name> -f <dockerfile> --squash .

If you require NVIDIA CUDA support, build the image with tag=cuda build argument.

$ docker build --build-arg tag=cuda -t <image-name>:cuda -f <dockerfile> .

⁠Run

Running the container without GPU support is straightforward.

$ docker run -it <image-name> /bin/bash

To run the container with GPU support, we recommend using nvidia-docker. The following example demonstrates the employment of two GPUs and executing bash.

$ docker run --runtime=nvidia \
    -e NVIDIA_VISIBLE_DEVICES=0,1 \
    -it <image-name>:cuda /bin/bash