TensorFlow Quick Start

Compute Resources

Docker Usage

Software Included

Interactive GUI Session

  • Interactions GUI sessions are done via the Custom noVNC Image application in Open On Demand (OOD).

  • You can find out more about OOD here: Compute Quick Start.

  • There are two fields beyond the basics that will need information specific to this image.
    • Environment Variables

    • Docker Image

Environment Variables

  • This information should be space separated in the field.

PASSWORD=password
  • Optional variables
    • GUI display size. This can be changed with the following variables.
      • Width default: 1024

      • Height default: 768

    DISPLAY_WIDTH=<width> DISPLAY_HEIGHT=<height>
    

Docker Image

ghcr.io/washu-it-ris/tensorflow:<tag>

TensorFlow Docker Tag

The <tag> will refer to the version of Tensorflow in the Docker container. Please click here see a current list of supported TensorFlow versions and their corresponding tags.

  • Fill out the rest of the fields with the appropriate information (explained in the quick start).

  • You will need to select a GPU in the field for doing so since TensorFlow uses GPUs.

  • Launch the job through the methods described in the quick start.

  • Once in an interactive TensorFlow session using the following command:

> ipython -i --no-banner

Interactive Command-Line Session

  • If you wish to use TensorFlow in an interactive command-line session, you can do so with the following command.

bsub -Is -R 'gpuhost' -gpu "num=1" -q general-interactive -a 'docker(ghcr.io/washu-it-ris/tensorflow:<tag>)' /bin/bash

Extend the TensorFlow Image

You may wish to extend the TensorFlow container with additional Python packages. Below is an example Dockerfile which extends the latest RIS-hosted TensorFlow image with the following packages:

Dockerfile

FROM ghcr.io/washu-it-ris/tensorflow:latest
RUN pip install pandas matplotlib
  • Please see this section for more information on building and pushing your own custom Docker container.

Available Versions

Current Version:

  • ghcr.io/washu-it-ris/tensorflow
    • latest, 24.04-tf2-py3

Earlier Versions:

  • gcr.io/ris-registry-shared/tensorflow
    • latest, 21.10-tf2-py3