OSG provides ready-to-use singularity containers for TensorFlow workflows for both CPU and GPU jobs . These containers are based on Ubuntu and live in cvmfs.
- /cvmfs/singularity.opensciencegrid.org/opensciencegrid/tensorflow:latest is the CPU only version of TensorFlow. This runs slower, but OSG has many more CPU only resources available than GPU resources.
- /cvmfs/singularity.opensciencegrid.org/opensciencegrid/tensorflow-gpu:latest is a modified TensorFlow-GPU image to be used on the GPUs available on OSG.
For CPU jobs, the following attributes need to be added to your submit file:
The tutorial example can be found in github here.
Working with TensorFlow and GPU resources interactively
The following link shows how to submit condor jobs, requesting GPU resources and how access them via SSH: Using GPU resources via SSH.